当前位置:   article > 正文

【caffe入门】在之前的模型上继续训练的两种方法_加入了--continue_train还是没有继续之前的训练

加入了--continue_train还是没有继续之前的训练

在caffe中有两种需要继续训练模型的情况:

  1. 训练突然被中断需要从之前的状态继续训练

            这种情况我们可以直接调用在中断前保存的solverstate文件就可以继续训练了,前提是要有记录的solverstate文件哦~

具体执行情况如下:

            之前的训练脚本 train.sh 如下:

  1. #!/usr/bin/env sh
  2. set -e
  3. TOOLS=/home/caffe/build/tools
  4. LOG=/home/myexample/logdata/log1.log
  5. $TOOLS/caffe train \
  6. --solver=compact_solver.prototxt 2>&1|tee $LOG

             在之前的状态上恢复训练的脚本train.sh如下:

  1. #!/usr/bin/env sh
  2. set -e
  3. TOOLS=/home/caffe/build/tools
  4. LOG=/home/myexample/logdata/log1.log
  5. $TOOLS/caffe train \
  6. --solver=compact_solver.prototxt \
  7. --snapshot=/home/myexample/models/compact_iter_3000.solverstate 2>&1|tee $LOG

              这里需要注意的是

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/419381
推荐阅读
  

闽ICP备14008679号