概述
错误缘由，在使用 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错误解决方法——简记所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape [9216,4096] an](/uploads/reation/bcimg2.png)
发表评论 取消回复