码迷,mamicode.com
首页 > 其他好文 > 详细

使用snapshot继续训练网络

时间:2017-11-27 23:48:33      阅读:394      评论:0      收藏:0      [点我收藏+]

标签:训练   bsp   examples   com   color   图片   tools   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中看起来这些行是分开的,实际上他们是通过"\"连接起来的,也就是说他们是一行。但我注释之后,

 

使用snapshot继续训练网络

标签:训练   bsp   examples   com   color   图片   tools   snapshot   脚本   

原文地址:http://www.cnblogs.com/ymjyqsx/p/7906149.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!