caffe命令列訓練以及測試
網易雲課堂(程式中檔案放在目錄~/Documents/caffe/projects/mnist中):https://study.163.com/course/courseLearn.htm?courseId=1003491001#/learn/video?lessonId=1004081127&courseId=1003491001
在終端中輸入sudo /home/caffe/build/tools/caffe
會出現許多幫助資訊
caffe後面可以跟4個命令:train、test、device_query、time
1、time:評估模型執行的時間大概是多少
命令(跑CPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10
指定模型以及迭代次數
命令(跑GPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10 -gpu 0
2、train:訓練網路
命令:sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt
出現錯誤:Unknown bottom blob 'label' (layer 'accuracy', bottom index 1)
將最後兩層,即accuracy和cost層註釋掉即可(問題:不註釋怎麼解決?)
將所有log都輸出到一個檔案中(螢幕沒有輸出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>1 | tee a.log
!!將所有log都輸出到一個檔案中(螢幕既有輸出,檔案也有輸出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>&1 | tee a.log
通過caffe的工具把log中有用的資訊提取出來,使用caffe/tools/extra/parse_log.py
另外一個工具更加直觀的,將log輸出圖,使用caffe/tools/extra/plot_training_log.py.example
輸出幫助資訊:
使用方法:sudo python /home/caffe/tools/extra/plot_training_log.py.example 4 plotlog.png a.log
生成了一張圖片plotlog.png,還有a.log.test和a.log.train兩個檔案
也可以用parse_log.py來生成a.log.test和a.log.train兩個檔案,然後自己進行畫圖
稍微有點問題
=====================分割線4-19======================
訓練時出現的問題,因為在定義網路的檔案中,即hbk_mnist.prototxt檔案中,在測試階段的資料層沒有label的輸出,修改後執行成功