有时候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
推荐博客