Caffe 分類問題 Check failed: error == cudaSuccess (2 vs. 0) out of memory

若是圖片過大,須要適當縮小batch_size的值,不然使用GPU時可能超出其緩存大小而報錯緩存

相關文章
相關標籤/搜索