ResourceExhaustedError错误解决方法——简记

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

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

  

 

  
1
 

解决方法如下：

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

文章来源: positive.blog.csdn.net，作者：墨理学AI，版权归原作者所有，如需转载，请联系作者。

原文链接：positive.blog.csdn.net/article/details/89499340