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

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

posted on 2018-05-23 11:31 NLazyo 阅读(1701) 评论(0) 编辑收藏举报