1. 程式人生 > >caffe呼叫之前的權重和接著斷點繼續訓練

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