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

相关文章