利用caffe的solverstate在意外断电退出时继续训练

来源:互联网 发布:mysql引擎是什么 编辑:程序博客网 时间:2024/06/05 22:50

caffe可以在遇到突然断开时候重新训练。在训练的时候不仅会保存当前模型的参数(也就是caffemodel)文件,也会把训练到当前状态信息全部保存下来,这个文件就是solverstate文件。只要在训练的时候加上snapshot参数就可以了

./build/tools/caffe train --solver=models/bvlc_reference_caffenet/solver.prototxt \

--snapshot=models/bvlc_reference_caffenet/caffenet_train_iter_10000.solverstate

tip:注意“\”,在ubuntu可以连着写的,换行写需要加上“\”。

阅读全文
0 0
原创粉丝点击