diff --git a/timm/optim/sgdp.py b/timm/optim/sgdp.py index 309ed69c..f4a94aa3 100644 --- a/timm/optim/sgdp.py +++ b/timm/optim/sgdp.py @@ -1,10 +1,11 @@ """ -AdamP -Copyright (c) 2020-present NAVER Corp. -MIT license +SGDP Optimizer Implementation copied from https://github.com/clovaai/AdamP/blob/master/adamp/sgdp.py Paper: `Slowing Down the Weight Norm Increase in Momentum-based Optimizers` - https://arxiv.org/abs/2006.08217 Code: https://github.com/clovaai/AdamP + +Copyright (c) 2020-present NAVER Corp. +MIT license """ import torch