Tensorflow exceeds 10% of system memory解决方法

25,558次阅读
没有评论
在训练一个模型的时候遇到这样的问题
Tensorflow exceeds 10% of system memory解决方法
在tensorflow的github上面也发现有人遇到类似的问题,贴上链接https://github.com/tensorflow/tensorflow/issues/18736

解决方法

Try reducing batch_size attribute to a small number
减小batch size,之前用的batch size比较大,后来调小一点就不会出现这样的问题
 1
admin
版权声明:本站原创文章,由 admin 2018-09-10发表,共计200字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)