如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错
Caffe 分类问题 Check failed: error == cudaSuccess (2 vs. 0) out of memory
最新推荐文章于 2021-04-19 14:47:36 发布
如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错