您的位置:首页 > Web前端

Caffe 分类问题 Check failed: error == cudaSuccess (2 vs. 0) out of memory

2018-05-23 11:31 483 查看

如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐