caffe训练resume

时间:2023-07-02 22:57:56
MODEL=${EXP}/model/${NET_ID}/pspnet101_VOC2012.caffemodel
SNAPSHOT=${EXP}/model/${NET_ID}/train_iter_7000.solverstate
CMD="${CAFFE_BIN} train \
--solver=${CONFIG_DIR}/solver.prototxt \
--gpu=${DEV_ID}"
#CMD="${CMD} --weights=${MODEL}" #train from init
CMD="${CMD} --snapshot=${SNAPSHOT}" #resume
echo Running ${CMD} && ${CMD} 2>&1 | tee train.log