使用snapshot继续训练网络

Posted 2020-10-15 去做点事情

注意：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中看起来这些行是分开的，实际上他们是通过"\\"连接起来的，也就是说他们是一行。但我注释之后，