caffe基础-10常见训练命令解析

caffe基础-10常见训练命令解析

###训练网络#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 2-weights:可选参数。用预先训练好的权重来fine-tuning模型，需要一个caffemodel如：# ./build/tools/caffe train -solver examples/finetuning_on_flickr_style/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_10000.caffemodel -gpu 0 -iterations 100time参数用来在屏幕上显示程序运行时间。如：# ./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