PyTorch学习之六个学习率调整策略.pdfVIP

  1. 1、本文档共2页,可阅读全部内容。
  2. 2、原创力文档(book118)网站文档一经付费(服务费),不意味着购买了该文档的版权,仅供个人/单位学习、研究之用,不得用于商业用途,未经授权,严禁复制、发行、汇编、翻译或者网络传播等,侵权必究。
  3. 3、本站所有内容均由合作方或网友上传,本站不对文档的完整性、权威性及其观点立场正确性做任何保证或承诺!文档内容仅供研究参考,付费前请自行鉴别。如您付费,意味着您自己接受本站规则且自行承担风险,本站不退款、不进行额外附加服务;查看《如何避免下载的几个坑》。如果您已付费下载过本站文档,您可以点击 这里二次下载
  4. 4、如文档侵犯商业秘密、侵犯著作权、侵犯人身权等,请点击“版权申诉”(推荐),也可以打举报电话:400-050-0827(电话支持时间:9:00-18:30)。
  5. 5、该文档为VIP文档,如果想要下载,成为VIP会员后,下载免费。
  6. 6、成为VIP后,下载本文档将扣除1次下载权益。下载后,不支持退款、换文档。如有疑问请联系我们
  7. 7、成为VIP后,您将拥有八大权益,权益包括:VIP文档下载权益、阅读免打扰、文档格式转换、高级专利检索、专属身份标志、高级客服、多端互通、版权登记。
  8. 8、VIP文档为合作方或网友上传,每下载1次, 网站将根据用户上传文档的质量评分、类型等,对文档贡献者给予高额补贴、流量扶持。如果你也想贡献VIP文档。上传文档
查看更多

PyTorch学习之六个学习率调整策略

PyTorch学习率调整策略通过torch.optim.lr_scheduler接⼝实现。PyTorch提供的学习率调整策略分为三⼤类,分别是

a.有序调整:等间隔调整(Step),按需调整学习率(MultiStep),指数衰减调整(Exponential)和余弦退⽕CosineAnnealing。

b.⾃适应调整:⾃适应调整学习率ReduceLROnPlateau。

c.⾃定义调整:⾃定义调整学习率LambdaLR。

1、等间隔调整学习率StepLR

等间隔调整学习率,调整倍数为gamma倍,调整间隔为step_size。间隔单位是step。需要注意的是,step通常是指epoch,不要弄成

iteration了。

torch.optim.lr_scheduler.StepLR(optimizer,step_size,gamma=0.1,last_epoch=-1)

step_size(int)-学习率下降间隔数,若为30,则会在30、60、90…个step时,将学习率调整为lr*gamma。

gamma(float)-学习率调整倍数,默认为0.1倍,即下降10倍。

last_epoch(int)-上⼀个epoch数,这个变量⽤来指⽰学习率是否需要调整。当last_epoch符合设定的间隔时,就会对学习率进⾏调整。当为-1

2、按需调整学习率MultiStepLR

按设定的间隔调整学习率。这个⽅法适合后期调试使⽤,观察loss曲线,为每个实验定制学习率调整时机。

torch.optim.lr_scheduler.MultiStepLR(optimizer,milestones,gamma=0.1,last_epoch=-1)

milestones(list)-⼀个list,每⼀个元素代表何时调整学习率,list元素必须是递增的。如milestones=[30,80,120]

gamma(float)-学习率调整倍数,默认为0.1倍,即下降10倍。

3、指数衰减调整学习率ExponentialLR

按指数衰减调整学习率,调整公式:lr=lr∗gamma∗∗epochlr=lr*gamma**epoch

torch.optim.lr_scheduler.ExponentialLR(optimizer,gamma,last_epoch=-1)

gamma-学习率调整倍数的底,指数为epoch,即gamma**epoch

4、余弦退⽕调整学习率CosineAnnealingLR

以余弦函数为周期,并在每个周期最⼤值时重新设置学习率。以初始学习率为最⼤学习率,以2∗T_max为周期,在⼀个周期内先下降,

后上升。

torch.optim.lr_scheduler.CosineAnnealingLR(optimizer,T_max,eta_min=0,last_epoch=-1)

T_max(int)-⼀次学习率周期的迭代次数,即T_max个epoch之后重新设置学习率。

eta_min(float)-最⼩学习率,即在⼀个周期中,学习率最⼩会下降到eta_min,默认值为0。

5、⾃适应调整学习率ReduceLROnPlateau

当某指标不再变化(下降或升⾼),调整学习率,这是⾮常实⽤的学习率调整策略。

例如,当验证集的loss不再下降时,进⾏学习率调整;或者监测验证集的accuracy,当accuracy不再上升时,则调整学习率。

torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer,mode=min,factor=0.1,patience=10,

verbose=False,threshold=0.0001,threshold_mode=rel,cooldown=0,min_lr=0,eps=1e-08)

mode(str)-模式选择,有min和max两种模式,min表⽰当指标不再降低(如监测loss),max表⽰当指标不再升⾼(如监测accuracy)。

factor(float)-学习率调整倍数(

文档评论(0)

梁学玉 + 关注
官方认证
文档贡献者

中专学生

认证主体宁夏三科果农牧科技有限公司
IP属地宁夏
统一社会信用代码/组织机构代码
91640500MABW4P8P13

1亿VIP精品文档

相关文档