您的位置:首页 > Web前端

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

2017-05-18 15:19 369 查看
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可以连着写的,换行写需要加上“\”。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: