caffe根目录下新建train.sh
./build/tools/caffe train --solver=models/VGGNet/VOC0712/SSD_300x300/solver.prototxt --weights=models/solver_iter_300000.caffemodel
生成log日志
cd caffe_Root
./build/tools/caffe train \
--solver="solver.prototxt" \
--gpu 0 2>&1 | tee path to log/name.log
使用tee进行写日志会使得当提前终止时,ctrl+c不能够保存当前的模型。tee改写了ctrl+c指令。
使用下面的方法保存日子,当使用CTRL+C时,可以保存当前训练得到的模型。
GLOG_logtostderr=0 GLOG_log_dir=/data_1/ssd-project/DSOD/caffe/models/DSOD300/bigobj_0617/DSOD300_300x300/
cd /data_1/ssd-project/DSOD/caffe
./build/tools/caffe train \
--solver="/data_1/ssd-project/DSOD/caffe/models/DSOD300/bigobj_0617/DSOD300_300x300/solver.prototxt" \
--snapshot="/data_1/ssd-project/DSOD/caffe/models/DSOD300/bigobj_0617/DSOD300_300x300/_iter_25.solverstate" \
--gpu 0