如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错
Caffe 分类问题 Check failed: error == cudaSuccess (2 vs. 0) out of memory
猜你喜欢
转载自www.cnblogs.com/bile/p/9076031.html
今日推荐
周排行