caffe训练resume

最新推荐文章于 2023-10-12 20:42:36 发布

weixin_33980459

最新推荐文章于 2023-10-12 20:42:36 发布

阅读量154

点赞数

文章标签：人工智能

原文链接：http://www.cnblogs.com/haiyang21/p/7533471.html

版权

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