天天看點

caffe随記(五)---/build/tools/caffe.bin工具簡析1、怎麼用這個指令2、caffe.cpp源檔案

在caffe根目錄下輸入如下指令:

 ./build/tools/caffe.bin, 得到如下顯示

caffe随記(五)---/build/tools/caffe.bin工具簡析1、怎麼用這個指令2、caffe.cpp源檔案

usage:caffe<command><args>

# 這個是告訴你使用格式, caffe 後接上 一個command指令,後面再接其他參數

commands: #你能選擇的指令有一下這麼幾種

train #訓練或者微調一個模型

test  #對一個模型打分

device—query #顯示GPU診斷資訊

time #評估模型執行時間

Flags form tools/caffe.cpp   #其他一些參數的總覽

-gpu         (可選;給定時運作GPU模式,用’ , ’分隔開不同的gpu,

                     ‘-gpu all’表示運作在所有可用的gpu裝置上,此時有效訓練批量大小就是gpu裝置數乘以batch_size)

-iterations (循環疊代次數,預設為50)

-level          (可選;定義網絡水準,也是NetState中的一個,但我也還不清楚這個的作用)

-model        (指定模型定義文本檔案名,xxx.prototxt)

-phase        (可選;網絡是處于TEST還是TRAIN階段,當你使用command中time指令時,再指定phase就可以選擇計算TEST或者TRAIN的耗時)

-sighup_effect (可選;當收到SIGHUP信号時要采取的動作,可選項:snapshot、stop、none,預設為snapshot,即列印快照)

-sigint_effect (可選;當收到當收到SIGINT信号時要采取的動作,可選項同上,預設stop)

-snapshot (可選,恢複訓練時指定上次中止的快照,就是比如訓練到一般按Ctrl+C終止訓練(Linux中這個Ctrl+C不是copy,而是終止目前操作),就會得到一個solverstate                        檔案,下次恢複訓練時就可以指定這個)

-solver       ( 指定sovler.prototxt檔案,在train的時候需要這個參數)

-stage        (可選;也是NetState中的一個,但我也還不清楚這個的作用)

-weights    ( 指定用于微調的預訓練權值,也即 訓練後得到的**.caffemodel檔案,不可與snapshot同時出現)

注:這個檔案的内容有些多,我也隻是選擇性的閱讀并注釋了部分。