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

有时候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

推荐博客

https://blog.csdn.net/hjxu2016/article/category/6625107

猜你喜欢

转载自blog.csdn.net/baidu_40840693/article/details/85337049