在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.sol

这篇博客介绍了在Caffe中如何从中断的训练状态恢复以及如何基于现有模型进行微调。对于训练中断的情况,可以使用solverstate文件继续训练,若训练参数未变,会接着上次的步数继续,否则从新步数开始。对于模型微调,可以通过修改脚本使用caffemodel文件进行训练,新方案会覆盖旧模型并按新配置迭代。

1万+

被折叠的 条评论
为什么被折叠?



