caffe: 使用.solverstate文件可以继续之前的训练

xiaoxiao2021-02-28  107

  我们在使用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   上述配置根据个人文件路径实际情况相应修改即可。

转载请注明原文地址: https://www.6miu.com/read-37450.html

最新回复(0)