caffe训练模型中断的解决办法(利用solverstate)

caffe训练过程中会生成.caffemodel和.solverstate文件,其中caffemodel为模型训练文件,可用于参数解析,solverstate为中间状态文件

当训练过程由于断电等因素中断时,可用solverstate文件继续执行,具体运行脚本和训练脚本类似,只需添加snapshot状态参数即可。

./build/tools/caffe train \

--solver=examples/test/solver.prototxt --snapshot=examples/test/test_100000.solverstate

posted @ 2019-01-17 14:22  老王哈哈哈  阅读(1225)  评论(0编辑  收藏  举报