【caffe入门】在之前的模型上继续训练的两种方法

最新推荐文章于 2025-05-17 12:20:03 发布

杨海er

最新推荐文章于 2025-05-17 12:20:03 发布

阅读量4k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： caffe

本文链接：https://blog.youkuaiyun.com/yh1226/article/details/85772198

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在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