- 博客(3)
- 资源 (2)
- 收藏
- 关注
原创 关于TensorFlow GPU 跑神经网络爆内存的处理办法
本人上一篇博客介绍了vgg16模型的finetuning(网址:点击打开链接),在cpu上跑没问题,但是我在全部复制到gpu(gtx750TI)上跑却发生爆内存的情况,下面介绍一下我的解决办法。 不得不承认爆内存跟我的渣渣gpu有关,为了尽量的减少gpu内存的使用率,在加载vgg16模型的时候并没有直接调用命令,而是把网络的权重跟偏置都设为常值,直接赋值,减少了以后sess.r...
2018-04-21 22:19:30 10981 1
原创 VGG16 finetuning实现猫狗分类
最近在学习VGG16网络的finetuning,想着在修改最后一层全连接层的基础上实现猫狗分类,但是在调试程序过程中发现了两个个好玩的东西:one—hot,权重部分载入。关于one-hot的基础介绍请参考:点击打开链接 个人认为这个one-hot在一些条件下也可以不用,关键取决于你全连接输出时有没有对全连接层实行softmax,若进行了softmax操作,则必须进行one-hot,若...
2018-04-14 10:34:36 2781 4
STATA创建空间权重矩阵及空间杜宾模型的命令.zip
2020-07-02
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人