我們在使用caffe訓練過程中會生成.caffemodel和.solverstate文件,一個是模型文件,一個是中間狀態文件(生成多少個取決於你自己設定的snapshot)。當訓練過程中斷,你想繼續運行數據學習,此時只需要調用.solverstate文件即可。
使用方式代碼,我使用的是.sh直接運行,配置和官方給的文件train_caffenet.sh差不多,稍微添加點內容就可以了。
./build/tools/caffe train \
--solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate
上述配置根據個人文件路徑實際情況相應修改即可。
如上圖可以看到~我是直接從第100000次的狀態開始執行的。
本文轉載自:http://blog.csdn.net/wang4959520/article/details/51831637