Resource exhausted: OOM when allocating tensor with shape[2] and type int32 on /job:localhost/replic

当显示Resource exhausted错误时,一般会在之前显示好多错误提示,提示的内容都和内存相关,此error是电脑内存不够,目前我找到解决问题的方法有三种:

1.改变输入图像的大小,例如原来输入图像的大小为448x448,你更改成224x224,再运行程序,当然需要修改你程序对应的部分。如果224x224还不可以的话,在不影响测试效果的情况下,继续减小,实在不行的话那就需要参考第三条了。

2.更改每次训练的batch_size,例如原来的batch_size=32,减小batch_size直到你的程序可以运行。

3.如果以上两种方法都不能解决,那就需要考虑更换电脑,加GPU了。

备注:在用anaconda自带的spyder运行程序时,程序卡了好长时间没有更新,是因为spyder不会将这些错误打印出来,会一直显示在运行程序给人错觉。这时可以使用pycharm或者在windows的Dos黑框下运行程序。

猜你喜欢

转载自blog.csdn.net/qq_22169787/article/details/89084495