From ec8836166e2e870d6719fbf6ad3de7e7547b954f Mon Sep 17 00:00:00 2001
From: Sangdoo Yun <oodgnas@gmail.com>
Date: Sat, 25 Jul 2020 12:43:51 +0900
Subject: [PATCH] Update sgdp.py

---
 timm/optim/sgdp.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/timm/optim/sgdp.py b/timm/optim/sgdp.py
index f713df1e..309ed69c 100644
--- 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