caffe调用之前的权重和接着断点继续训练

最新推荐文章于 2025-06-03 10:44:45 发布

原创最新推荐文章于 2025-06-03 10:44:45 发布 · 4.4k 阅读

3 ·

CC 4.0 BY-SA版权

深度框架|caffe 专栏收录该内容

47 篇文章

订阅专栏

本文介绍如何在Caffe中续接已有的训练任务。包括使用snapshot文件继续训练及加载预训练模型进行二次训练的方法。

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

有时候caffe在训练的时候，因某种原因不得不暂停训练，下次怎么接着继续训练呢，下列sh脚本文件

#!/usr/bin/env sh

Path=/home/ccf/CCF/data-me
/home/hjxu/caffe-master/caffe/build/tools/caffe train --solver=/home/hjxu/WSI-metastic/data-me/profile/solver.prototxt  \
--snapshot=/home/hjxu/WSI-metastic/data-me/model/caffe_Breast_transfer_train_iter_95000.solverstate 2>&1|tee /home/hjxu/WSI-metastic/data-me/train2.txt

如果已经有了模型，想要第二次加强训练

#!/usr/bin/env sh

Path=/home/ccf/CCF/data-me
/home/hjxu/caffe-master/caffe/build/tools/caffe train --solver=/home/hjxu/WSI-metastic/data-me/profile/solver.prototxt  \
--weights=/home/hjxu/WSI-metastic/data-me/model/caffe_Breast_transfer_train_iter_95000.caffemodel 2>&1|tee /home/hjxu/WSI-metastic/data-me/train2.txt