Check failed: error == cudaSuccess (2 vs. 0) out of memory

finitune  googlenet時,train未報錯,test時報了, stackflow裏面的回答 1.減小mini_batch 2.減小圖片大小 3.升級GPU(stackflow裏面一般都是8G-12G的,別的CS的實驗室好像直接就用的GTX 1080,可以考慮)
相關文章
相關標籤/搜索