GPU: (1050i)
real 1m25.823s user 1m10.579s sys 0m19.228sCPU:
real 11m25.795s user 11m28.551s sys 0m0.444s
git clone caffe 下來,在 example folder 有 mnist。
先 download data:
./data/mnist/get_mnist.sh ./examples/mnist/create_mnist.sh開始 run train in..
./examples/mnist/train_lenet.shrun 完,結果會是在 example/mnist 下..lenet_iter_*
修改 example/mnist/lenet_solver.prototxt,最後一行 solver_mode: GPU 改 CPU,就可以強至用 CPU training
caffe 附的一些 tool,可以用來看 training 的 trainning error ..
把 train 時的 log 紀錄到一個 file:
./examples/mnist/train_lenet.sh 2>&1 | tee -a lenet然後用 parse_log.py 整理一下...
./tools/extra/parse_log.py lenet .會產生 lenet.test, lenet.train
寫 plotcmd,教給 gnuplot 執行...
set datafile separator ',' set term x11 0 plot './lenet.train' using 1:4 with line,\ './lenet.test' using 1:5 with line用 gnuplot 畫出來..
gnuplot -persist plotcmd
使用 GTX-950M:
real 2m32.821s user 1m53.398s sys 0m43.228s
沒有留言:
張貼留言