红联Linux门户
Linux帮助

caffe中fine tuning时的sh命令

发布时间:2016-10-12 11:29:55来源:cnblogs.com/darkknightzh作者:darkknightzh
1.如果直接训练时,test.sh中内容如下:
./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt --gpu 1
之后在caffe根目录输入如下命令:
sh examples/XXX/test.sh
 
2.如果使用预训练好的模型进行fine tuning,则test.sh中内容如下:
./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt -weights examples/models/finetuning.caffemodel --gpu 1
其中finetuning.caffemodel位于examples/models/文件夹内。
之后在caffe根目录使用sh examples/XXX/test.sh即可。
 
参考:
caffe中fine tuning时的sh命令
 
本文永久更新地址:http://www.linuxdiyf.com/linux/24963.html