Caffe模型训练之solver.prototxt配置

在caffe框架中模型训练之前,需要配置solver.prototxt文件做一些配置,现对其做一些讲解:
1.test_iter
test_iter=测试集大小/测试层中batch_size
2.max_iter
max_iter=test_iter*80,其中80是经验值,你可以调大或调小一些,总之使在max_iter次迭代训练中,模型要收敛即可
3.test_interval
test_interval=训练集大小/训练层batch_size大小
4.base_lr
一般保持原始值不变,也可以适当调整
5.solver_mode
若是用CPU训练则写为CPU,若是用GPU跑模型,则写为GPU;GPU比CPU快很多,难以想象

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值