caffe中断后,如何继续在之前的模型上训练

两种方法 :
一:使用solverstate文件继续训练.
继续之前的迭代次数进行训练.如之前在迭代200次中断的,继续训练后在从200开始继续,200,210,230....
#!/bin/bash
LOG=examples/VOCFLIR/out/log/my_train_lite-`date +%Y-%m-%d-%H-%M-%S`.log
./build/tools/caffe train --solver examples/VOCFLIR/my_mobilenet_yolov3_lite_solver.prototxt --snapshot examples/VOCFLIR/out/models/my_mobilenet_yolov3_lite_deploy_iter_24000.solverstate    --gpu=0 2>&1 | tee $LOG

二:使用caffemodel 文件继续训练.
相当于使用预训练模型,迭代次数从0开始.如之前在迭代200次中断的,继续训练后在从0开始继续,10,20,30....
#!/bin/bash
LOG=examples/VOCFLIR/out/log/my_train_lite-`date +%Y-%m-%d-%H-%M-%S`.log
./build/tools/caffe train --solver examples/VOCFLIR/my_mobilenet_yolov3_lite_solver.prototxt   --weights  examples/VOCFLIR/out/models/my_mobilenet_yolov3_lite_deploy_iter_24000.caffemodel    --gpu=0 2>&1 | tee $LOG
 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值