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

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

猜你喜欢

转载自www.cnblogs.com/bile/p/9076031.html