错误缘由,在使用 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 配置文件中进行设置,如下图:

最后
以上就是温柔冬瓜最近收集整理的关于ResourceExhaustedError错误解决方法——简记的全部内容,更多相关ResourceExhaustedError错误解决方法——简记内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复