我们在使用caffe训练过程中会生成.caffemodel和.solverstate文件,一个是模型文件,一个是中间状态文件(生成多少个取决于你自己设定的snapshot)。当训练过程中断,你想继续运行数据学习,此时只需要调用.solverstate文件即可。
使用方式代码,我使用的是.sh直接运行,配置和官方给的文件train_caffenet.sh差不多,稍微添加点内容就可以了。
./build/tools/caffe train \
--solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate
上述配置根据个人文件路径实际情况相应修改即可。
如上图可以看到~我是直接从第100000次的状态开始执行的。
本文转载自:http://blog.csdn.net/wang4959520/article/details/51831637