bug6-ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128]

1147-柳同学

发表文章数:589

热门标签

,
首页 » 算法 » 正文

完整错误:

tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating
 tensor with shape[512,15,15,128] and type float on /job:localhost/replica:0/task:0
 /device:GPU:0 by allocator GPU_0_bfc [Op:Conv2D]

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

标签:

未经允许不得转载:作者:1147-柳同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《bug6-ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128]》 发布于2020-10-12

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录