在windows,linux平台进行caffemodel的批量训练

笔者之前在训练caffemodel的时候,都是先设置好一个网络模型,然后再在命令行中输入训练执行命令比如(linux中):

./build/tools/caffe train --solver=examples/mydata/huihua_dongfang/mynetV2/solver.prototxt

后来发现写一个批处理程序,可以一下子处理很多个网络模型,非常简单高效,能将自己从简单劳动中解放出来。具体的我分为windows, linux两个系统中分别讲解。


1,linux系统中,批量训练4个网络模型, 新建一个shell脚本文件,命名为“train.sh”, 在里面输入训练模型执行命令:

./build/tools/caffe train --solver=examples/mydata/huihua_dongfang/mynetV2/solver.prototxt
./build/tools/caffe train --solver=examples/mydata/huihua_dongfang/mynetV3/solver.prototxt
./build/tools/caffe train --solver=examples/mydata/huihua_dongfang/mynetV4/solver.prototxt
./build/tools/caffe train --solver=examples/mydata/huihua_dongfang/mynetV5/solver.prototxt


这样,在mynetV2训练好之后,会自动切换到mynetV3进行训练,非常简单高效。


2,windows系统中,批量训练4个网络模型, 新建一个bat批处理文件,命名为“train.bat”,在里面输入训练模型执行命令:

"caffe/caffe.exe" train --solver=V1211solver-48.prototxt --weights=train_V9_64/models_48_1/_iter_220000.caffemodel >log/V12_11.log 2>&1
"caffe/caffe.exe" train --solver=V1212solver-48.prototxt --weights=train_V9_64/models_48_1/_iter_220000.caffemodel >log/V12_12.log 2>&1
"caffe/caffe.exe" train --solver=V1213solver-48.prototxt --weights=train_V9_64/models_48_1/_iter_220000.caffemodel >log/V12_13.log 2>&1

这样程序自动在train_V9_64/models_48_1/_iter_220000.caffemodel的基础上进行finetune,并将训练日子保存到 log/V12_11.log 中, 第一个模型训练好之后,自动切换到第一个模型。


这篇博客,对于模型的批量训练非常实用,推荐给大家!


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值