電腦跑卷積神經網路 總是卡 有沒有比較好的解決方法?
剛開始學卷積神經網路 用tensorflow 寫的四層 兩層卷積加池化 兩層Fc 用Spyder CPU內存12G i5 7代 用的mnist數據 電腦總是跑卡 內存佔用100% 有沒有比較好的解決辦法?
- 內存佔滿換顯卡也沒用,顯存同一級別基本很少比內存更大的。 一個可能是每次所有樣本一起訓練,建議用分batch訓練。
- 如果分batch訓練的話,調整batchsize大小,比如從128調為10,極限為1
- 如果還卡,估計就是參數太多,比如一個圖片就50M,那就重新規劃訓練樣本。
換電腦吧
如果能正常運行,那代碼應該沒問題,卡可能是電腦配置的問題。而且對於cnn一般用GPU運行會合理一些。可以試試在智星雲分時租用一下GPU,看看跑的效果。