使用snapshot继续训练网络

注意:snapshots和weights不能同时使用

用预训练模型进行finetune是以下命令:

./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt -weights examples/models/finetuning.caffemodel --gpu 1

使用快照文件(.solverstate),则可以使用-snapshot参数:

 

./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt -snapshot examples/xxx/snapshot/xxx_iter_5000.solverstate --gpu 1

注意:sensenet也可以这样训练

在用sensenet v9训练数据时出现以下错误:

这是原本的shell脚本,是利用pretrain来训练

现在我想使用快照来恢复之前的训练,所以我把weights注释掉了,直接增加了snapshot。

但是一直报以下的错误:

就是没有solver的意思,但是我的solver文件一直在那,我并没有动。原因是:在vim中看起来这些行是分开的,实际上他们是通过"\"连接起来的,也就是说他们是一行。但我注释之后,

 

posted @ 2017-11-27 21:08  有梦就要去实现他  阅读(1493)  评论(0编辑  收藏  举报