caffe训练过程中会生成.caffemodel和.solverstate文件,其中caffemodel为模型训练文件,可用于参数解析,solverstate为中间状态文件
当训练过程由于断电等因素中断时,可用solverstate文件继续执行,具体运行脚本和训练脚本类似,只需添加snapshot状态参数即可。
./build/tools/caffe train \
--solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:caffe训练模型中断的解决办法(利用solverstate) - Python技术站