caffe训练resume

 

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

 

posted @ 2017-09-16 23:32  BlueOceans  阅读(445)  评论(0)  编辑  收藏  举报