caffe的solverstate的使用

   我们在使用caffe训练过程中会生成.caffemodel和.solverstate文件,一个是模型文件,一个是中间状态文件(生成多少个取决于你自己设定的snapshot)。当训练过程中断,你想继续运行数据学习,此时只需要调用.solverstate文件即可。

使用方式代码,我使用的是.sh直接运行,配置和官方给的文件train_caffenet.sh差不多,稍微添加点内容就可以了。

./build/tools/caffe train \
     --solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate

上述配置根据个人文件路径实际情况相应修改即可。

caffe的solverstate的使用_第1张图片

如上图可以看到~我是直接从第100000次的状态开始执行的。

你可能感兴趣的:(caffe)