tensorflow在显存不足的情况下,怎样使用较大的batch进行训练
楼主的问题解决了吗?我也遇到了类似问题,求解决办法。
■网友
假如你的pcie带宽足够,你可以把参数放在cpu上,梯度在cpu上更新。
■网友
【tensorflow在显存不足的情况下,怎样使用较大的batch进行训练】 我也在找一个类似caffe下iter_size的方法 难道要自己写.
____添加:
how to implement iter_size like caffe in tensorflow
建议是用多卡替代iter_size吧,毕竟硬件解决是最简单的解决方法。如果单卡显存只有2G,那优化也是很困难的(稍大一点的模型都跑不起来)。
方案:推荐使用slim吧,简单使用多GPU
■网友
使用更少的层,更小的单个样本大小,channel由3纬度变为1.或使用squeeze net等,压缩模型
推荐阅读
- 广汽传祺GM8|这车颜值超高,不足17万,390牛·米,动力强过GL8,气场不输埃尔法
- 汽车知识|不该被忽视的“中国制造”,配7座十年不大修,不足5万反成笑话!
- 大众高尔夫|没错,大众全新数字高尔夫来了,MQB Evo平台,入门级不足13万
- 老易聊车|这代步车再度刷新纪录!三个月销量近7万台,不足3万就能买
- Tensorflow 中咋定义自己的层呢
- 汽车知识|今日上市,轴距2米7比H6漂亮10倍,不足7万开出17万档次
- 汽车知识|又一亲民“撩妹王”来了!比思域还高级,不足8万,还看啥卡罗拉
- 京东商城存在着哪些不足
- 汽车爱卿|配爱信6AT+8.6s破百,远销36个国家,品牌还是来源英国,不足8w起
- 长安|实拍:长安版“雷克萨斯”,全新中级车配1.5T动力,不足8万
