進行模型訓練時,常出現 Check failed: error == cudaSuccess (2 vs. 0) out of memory??
uj5u.com熱心網友回復:
極有可能是batch_size的值設定太大,一次性讀入的圖片太多了,所以就超出了顯存。因此需要將train.prototxt中的檔案train和test的batch_size調小一點。轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/279014.html
標籤:Qualcomm開發
上一篇:1x1卷積核有啥具體的作用??
下一篇:碳基半導體優勢是什么?
