码迷,mamicode.com
首页 > 编程语言 > 详细

''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[?]'' 错误分析

时间:2018-06-21 22:36:47      阅读:910      评论:0      收藏:0      [点我收藏+]

标签:总结   连接   解决方法   维度   batch   err   出现   bsp   图片   

这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。

解决方法就是降低显卡的使用内存,途径有以下几种措施:

1 减少Batch 的大小

2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的

3 增加pool 层,降低整个网络的维度。

4 修改输入图片的大小

 

总结以下:想进一切办法降维,降低网络的维度。

''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[?]'' 错误分析

标签:总结   连接   解决方法   维度   batch   err   出现   bsp   图片   

原文地址:https://www.cnblogs.com/code-wangjun/p/9210864.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!