CAFFE編譯測試的小總結

1.   搭建的環境和代碼:win7 64bit + vs2013+CUDA7.5

 http://blog.csdn.net/thesby/article/details/50880802


2.  編譯,製作數據集,訓練參考的:

 http://www.cnblogs.com/denny402/tag/caffe/

http://linusp.github.io/2015/07/21/caffe-base-usage.html

http://blog.csdn.net/u013657981/article/details/49497753


遇到的問題:

遇到很多很多問題。

如:1.  Check failed: error == cudaSuccess (35 vs. 0) CUDA driver version is insufficient for CUDA runtime version

解決辦法:換英偉達顯卡版本更高的電腦訓練模型。

參考地址:http://stackoverflow.com/questions/31688759/mac-caffe-cuda-driver-issue

2.“error == cudaSuccess (2 vs. 0) out of memory”

解決辦法:顯存不足的錯誤,把train_val.prototxt中的batch_size從256一路改到4才成功

參考地址:http://blog.csdn.net/songyu0120/article/details/46728113

3. 在訓練模型的時候,無故就奔潰了:

解決辦法:是因爲被訓練的圖片大小是:136*38,尺寸太小。我最後把它擴展成256*256,模型訓練成功。

4. 訓練出來的模型,分類的分不出來:

這個問題,糾結了兩天,解決了。

主要問題是: 它在做模型的時候沒有收斂。 導致分類不出來。如何才能讓它收斂呢:

a. 因爲我的GPU性能差,所以我換成同事的電腦,是GFORCE 950.

b. 即使是950還是不行。 因爲我把batch_size改爲了4,太小了。導致其不收斂。

c. 最後我把solver.prototxt的編譯model從GPU改爲了CPU。 將batch_size改爲256, test的batch_size改爲了50,模型訓練了一個晚上,收斂了。




發佈了236 篇原創文章 · 獲贊 82 · 訪問量 140萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章