我是靠谱客的博主 温柔冬瓜,这篇文章主要介绍ResourceExhaustedError错误解决方法——简记,现在分享给大家,希望可以做个参考。

错误缘由,在使用 tensorflow-gpu 进行图像生成模型训练时,报错如下:

    ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[16,64,224,224]

解决方法如下:

  • [16,64,224,224] 的第一个参数表示batch_size的大小,第二个参数表示某层卷积核的个数,第三个参数表示图像的长,第四个参数表示图像的宽
  • 这里出现错误为内存超出,可以通过适当减小batch_size的大小,即可解决
  • 我这里把,batch_size大小由64修改为4,则该错误消失。
  • 我的batch_size变量在项目的config.py 配置文件中进行设置,如下图:

1


最后

以上就是温柔冬瓜最近收集整理的关于ResourceExhaustedError错误解决方法——简记的全部内容,更多相关ResourceExhaustedError错误解决方法——简记内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(127)

评论列表共有 0 条评论

立即
投稿
返回
顶部