1. caffeをインストールします(ubuntu 18.04)
sudo apt-get install caffe-cuda
2.例
-
データtrainlist.txtを準備する
-
train.prototxt(データレイヤー、つまりtrainlistを含む)
-
solver.prototxt(トレーニングパラメーターの設定など)
コマンド
caffe train -solver ./solver.prototxt -gpu 0 2>&1 | tee ./log/log_new.txt
#cpu
#caffe train -solver ./solver.prototxt 2>&1 | tee ./log/log.txt
3. caffeのtrain.prototxt Netscopeの視覚化
netscope公式ウェブサイトhttp://ethereon.github.io/netscope/quickstart.html
- URLを入力してください
- train.prototxtのネットワーク設計コードをURLに貼り付け、Shift + Enter