caffe訓練模型中斷的解決辦法(利用solverstate)


caffe訓練過程中會生成.caffemodel和.solverstate文件,其中caffemodel為模型訓練文件,可用於參數解析,solverstate為中間狀態文件

當訓練過程由於斷電等因素中斷時,可用solverstate文件繼續執行,具體運行腳本和訓練腳本類似,只需添加snapshot狀態參數即可。

./build/tools/caffe train \

--solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM