Update sgdp.py

5 years ago · ec8836166e
parent fceb09a44c
commit ec8836166e
1 changed files with 3 additions and 0 deletions
--- a/timm/optim/sgdp.py
+++ b/timm/optim/sgdp.py
@ -2,6 +2,9 @@
 AdamP
 Copyright (c) 2020-present NAVER Corp.
 MIT license
+
+Paper: `Slowing Down the Weight Norm Increase in Momentum-based Optimizers` - https://arxiv.org/abs/2006.08217
+Code: https://github.com/clovaai/AdamP
 """

 import torch