赞
踩
在caffe中有两种需要继续训练模型的情况:
这种情况我们可以直接调用在中断前保存的solverstate文件就可以继续训练了,前提是要有记录的solverstate文件哦~
具体执行情况如下:
之前的训练脚本 train.sh 如下:
- #!/usr/bin/env sh
- set -e
-
- TOOLS=/home/caffe/build/tools
- LOG=/home/myexample/logdata/log1.log
- $TOOLS/caffe train \
- --solver=compact_solver.prototxt 2>&1|tee $LOG
在之前的状态上恢复训练的脚本train.sh如下:
- #!/usr/bin/env sh
- set -e
-
- TOOLS=/home/caffe/build/tools
- LOG=/home/myexample/logdata/log1.log
- $TOOLS/caffe train \
- --solver=compact_solver.prototxt \
- --snapshot=/home/myexample/models/compact_iter_3000.solverstate 2>&1|tee $LOG
这里需要注意的是
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。