pytorch之warm-up预热学习策略(学习率)

Warmup: 在训练开始的时候先选择使用一个较小的学习率,训练了一些epoches或者steps(比如4个epoches,10000steps),再修改为预先设置的学习来进行训练。

Warmup使用原因:

由于刚开始训练时,模型的权重(weights)是随机初始化的,此时若选择一个较大的学习率,可能带来模型的不稳定(振荡),选择Warmup预热学习率的方式,可以使得开始训练的几个epoches或者一些steps内学习率较小,在预热的小学习率下,模型可以慢慢趋于稳定,等模型相对稳定后再选择预先设置的学习率进行训练,使得模型收敛速度变得更快,模型效果更佳。  

六个学习率调整策略:

  • (1)  等间隔调整学习率 StepLR
  • (2)按需调整学习率 MultiStepLR
  • (3)指数衰减调整学习率 ExponentialLR
  • (4)余弦退火调整学习率 CosineAnnealingLR
  • (5)自适应调整学习率 ReduceLROnPlateau
  • (6)自定义调整学习率 LambdaLR

转载:pytorch之warm-up预热学习策略_pytorch warmup-CSDN博客

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值