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

若是图片过大,须要适当缩小batch_size的值,不然使用GPU时可能超出其缓存大小而报错缓存

相关文章
相关标签/搜索