我们在使用caffe训练过程中会生成.caffemodel和.solverstate文件,一个是模型文件,一个是中间状态文件(生成多少个取决于你自己设定的snapshot)。当训练过程中断,你想继续运行数据学习,此时只需要调用.solverstate文件即可。
使用方式代码,我使用的是.sh直接运行,配置和官方给的文件train_caffenet.sh差不多,稍微添加点内容就可以了。
[plain]
view plain
copy
sudo GLOG_logtostderr=0 GLOG_log_dir='examples/128/models/2_1/' /build/tools/caffe train -solver=examples/128/solver.prototxt --snapshot=examples/128/test_100000.solverstate
上述配置根据个人文件路径实际情况相应修改即可。