Caffe的solverstate的使用

我們在使用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

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