L1aoXingyu/code-of-learn-deep-learning-with-pytorch

权重衰减可以通过 torch.optim.lr_scheduler 的相关类完成,从而使权重衰减和训练逻辑分离

xiaoyuan0203 opened this issue · 0 comments

@L1aoXingyu

作者你好,非常感谢您的分享。这里我提出一点小小的意见。

权重衰减可以通过 torch.optim.lr_scheduler 的相关类如 ExponentialLR完成,从而使权重衰减和训练逻辑分离。作权重衰减相关介绍的时候可以这么来介绍(实际使用也是这么使用比较广泛)。

https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate