在caffe中有两种需要继续训练模型的情况:
- 训练突然被中断需要从之前的状态继续训练
这种情况我们可以直接调用在中断前保存的solverstate文件就可以继续训练了,前提是要有记录的solverstate文件哦~
具体执行情况如下:
之前的训练脚本 train.sh 如下:
#!/usr/bin/env sh
set -e
TOOLS=/home/caffe/build/tools
LOG=/home/myexample/logdata/log1.log
$TOOLS/caffe train \
--solver=compact_solver.prototxt 2>&1|tee $LOG
在之前的状态上恢复训练的脚本train.sh如下:
#!/usr/bin/env sh
set -e
TOOLS=/home/caffe/build/tools
LOG=/home/myexample/logdata/log1.log
$TOOLS/caffe train \
--solver=compact_solver.prototxt \
--snapshot=/home/myexample/models/compact_iter_3000.solverstate 2>&1|tee $LOG
这里需要注意的是