让一个 caffe 跑起来的流程及注意事项

需要三个文件,以mnist为例,有这三个 train_lenet.sh,lenet_solver.prototxt,lenet_train_test.prototxt。
总的来说,执行 train_lenet.sh —> lenet_solver.prototxt —> lenet_train_test.prototxt,再开始训练网络。

①其中, train_lenet.sh是自己定义的脚本文件,内容如下:
/home/justin/caffe/build/tools/caffe train -solver lenet_solver.prototxt
-sigint_effect snapshot 2>&1 | tee -a ./lenet/solve.log

注意路径,上面的脚本要和lenet_solver.prototxt放一个目录下,脚本才能调用lenet_solver.prototxt。
./lenet/solve.log 意思是在当前目录下的lenet文件夹,保存训练的log文件。
2>&1的解释:Linux的输出中 1 表示标准输出,2 表示错误输出,2>&1 表示将错误的输入重定向到标准的输出中去。(这个不是很懂具体的意思)

②其中,lenet_solver.prototxt中注意

net: "lenet_train_test.prototxt"
snapshot_prefix: "lenet/"

lenet_solver.prototxt要和lenet_train_test.prototxt在一个目录下,lenet_solver.prototxt才能调用lenet_train_test.prototxt。
snapshot_prefix在train_lenet.sh和lenet_solver.prototxt中都要定义,不然会报错。

③其中,lenet_train_test.prototxt定义的是网络结构,这里要注意数据的路径

 source: "./mnist_train_lmdb"

上面的意思是当前路径的mnist_train_lmdb存放了训练数据。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值