训练网络
#sudo sh ./build/tools/caffe train --solver=examples/mnist/train_lenet.sh
-solver:必选参数。一个protocol buffer类型的文件,即模型的配置文件。如:
# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt
-gpu:可选参数。改参数用来指定哪一块gpu运行,根据gpu的id进行选择,如果设置为‘-gpu all’ 则使用的gpu运行。如果使用第二块gpu运行:
# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 1
-weights:可选参数。使用预训练好的权重来fine-tuning模型,需要一个caffemodel如:
# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel
测试网络
#./build/tools/caffe test -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_1000.caffemodel -gpu0 -iterations 100
time参数用来显示屏幕上显示程序运行时间。如:
# ./build/tools/caffe time -model examples/mnist/lenet_train_test.prototxt -iterations 10
关于gpu
# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 0,1
# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all