当前位置:   article > 正文

torch.optim.lr_scheduler.MultiStepLR_torch.optim.lr_scheduler.multisteplr milestones

torch.optim.lr_scheduler.multisteplr milestones

torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones, gamma=0.1, last_epoch=-1)
milestones为一个数组,如 [50,70]. gamma为倍数。如果learning rate开始为0.01 ,则当epoch为50时变为0.001,epoch 为70 时变为0.0001。
当last_epoch=-1,设定为初始lr。

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/328450
推荐阅读
相关标签
  

闽ICP备14008679号