Caffe之learning rate policy

learning rate很重要,如何设置它有很多种方法,在Caffe源码的caffe-master\src\caffe\solvers\sgd_solver.cpp中的GetLearningRate函数注释中有对应的介绍,如下:

// policies are as follows:
//    - fixed: always return base_lr.
//    - step: return base_lr * gamma ^ (floor(iter / step))
//    - exp: return base_lr * gamma ^ iter
//    - inv: return base_lr * (1 + gamma * iter) ^ (- power)
//    - multistep: similar to step but it allows non uniform steps defined by
//      stepvalue
//    - poly: the effective learning rate follows a polynomial decay, to be
//      zero by the max_iter. return base_lr (1 - iter/max_iter) ^ (power)
//    - sigmoid: the effective learning rate follows a sigmod decay
//      return base_lr ( 1/(1 + exp(-gamma * (iter - stepsize))))
//

废话不多说,直接看它们的图像一目了然:


对应matlab代码是:

iter=1:50000;  
max_iter=50000;  
base_lr=0.01;  
gamma=0.0001;  
power=0.75;  
step_size=5000;  
% - fixed: always return base_lr.  
lr=base_lr*ones(1,50000);  
subplot(2,3,1)  
plot(lr)  
title('fixed')  
% - step: return base_lr * gamma ^ (floor(iter / step))  
lr=base_lr .* gamma.^(floor(iter./10000));  
subplot(2,3,2)  
plot(lr)  
title('step')  
% - exp: return base_lr * gamma ^ iter  
lr=base_lr * gamma .^ iter;  
subplot(2,3,3)  
plot(lr)  
title('exp')  
% - inv: return base_lr * (1 + gamma * iter) ^ (- power)  
lr=base_lr.*(1./(1+gamma.*iter).^power);  
subplot(2,3,4)  
plot(lr)  
title('inv')  
% - multistep: similar to step but it allows non uniform steps defined by  
% stepvalue  
% - poly: the effective learning rate follows a polynomial decay, to be  
% zero by the max_iter. return base_lr (1 - iter/max_iter) ^ (power)  
lr=base_lr *(1 - iter./max_iter) .^ (power);  
subplot(2,3,5)  
plot(lr)  
title('poly')  
% - sigmoid: the effective learning rate follows a sigmod decay  
% return base_lr ( 1/(1 + exp(-gamma * (iter - stepsize))))  
lr=base_lr *( 1./(1 + exp(-gamma * (iter - step_size))));  
subplot(2,3,6)  
plot(lr)  
title('sigmoid')

What is `lr_policy` in Caffe?



  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值