从零开始学caffe(十):caffe中snashop的使用

在caffe的训练期间,我们有时候会遇到一些不可控的以外导致训练停止(如停电、设备故障灯),我们就不得不重新开始训练,这对于一些大型项目而言是非常致命的。在这里,我们介绍一些caffe中的snashop。利用snashop我们就可以实现训练的继续进行。
在这里插入图片描述
在之前我们训练得到的文件中,我们发现有后缀分别为caffemodel和solverstate的两种文件,其中caffemodel我们知道是训练的模型,而solverstate则是记录训练过程中的一系列参数,在这里,我们即可以用这个solverstate文件继续进行训练。
这里需要注意的一点是,我们首先需要进入超参数文件将其中最大迭代次数改为4000,因为在之前已经完成了3000次的最大迭代次数,模型训练已经结束。
我们将之前的批处理文件做如下修改:

%train训练数据%
%超参数文件%

E:\caffe-windows\Build\x64\Release\caffe.exe train ^
-solver=E:/caffe-windows/models/my_models_recognition/solver.prototxt ^
-snapshot=E:/caffe-windows/models/my_models_recognition/model/_iter_3000.solverstate
pause

运行批处理文件,我们发现模型从3000次开始继续训练,说明snashop发挥出它的作用。在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值