tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shap

报错:

tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[64,147,147,64]

出现以上类似的错误,主要是因为模型中的batch_size值设置过大,导致内存溢出,batch_size是每次送入模型中的值,由于GPU的关系,一般设为16,32,64,128。


阅读更多
文章标签: tensorflow报错
个人分类: TensorFlow
上一篇eclipse无法解析导入 java.util
下一篇并行编程模型opencl、mpi、cuda等的区别
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭