caffe呼叫之前的權重和接著斷點繼續訓練
阿新 • • 發佈:2018-12-29
有時候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
推薦部落格