Tensorflow exceeds 10% of system memory解决方法

26,731次阅读
没有评论

共计 200 个字符,预计需要花费 1 分钟才能阅读完成。

在训练一个模型的时候遇到这样的问题
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比较大,后来调小一点就不会出现这样的问题
正文完
请博主喝杯咖啡吧!
post-qrcode
 1
admin
版权声明:本站原创文章,由 admin 2018-09-10发表,共计200字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)
验证码