caffe.exe (caffe.bin)用法回顧


caffe.bin :command line brew

usage : caffe  <command><args>

commands:

  train:         訓練或者微調一個網絡

  test:        對一個模型打分測試 

  device_query:     顯示GPU診斷信息

  time:          評估模型執行時間

 

Flags from tools/caffe.cpp:

  -gpu (可選參數,給定時運行在GPU模式,‘-gpu all’ 則表示運行在所有可用的GPU設備上,此時真正訓練批量大小是N×B,N為指定的GPU設備的數目)

  -iterations (循環迭代次數,默認為50)

  -model  (指定模型定義文本名,*.prototext)

  -sighup_effect (當收到SIGHUP信號時要采取的動作,可選項:snapshot.stop,或none,默認是snapshot,即拍快照)

  -sigint_effect (當收到SIGINT信號時要采取的動作,可選項同上,默認為stop)

  -snapshot  (恢復訓練時所需要指定上次終止的快照,*.solvestate)

  -solver  (指定求解器文本文件嗎,*.prototxt)

  -weights (指定用於微調的與訓練權值,*.caffemodel,不可與snapshot同時出現)

 

example mnist:

train: caffe train --solve = example/mnist/lenet_solver.prototxt

test:  caffe test -model example/mnist/lenet_train_test.prototxt -weights example/mnist/lenet_iter_10000.caffemodel -iterations 100.


免責聲明!

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



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