Update adamp.py

pull/195/head
Sangdoo Yun 5 years ago committed by GitHub
parent 0e4bbb3bba
commit ec9e7533a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,10 +1,11 @@
"""
AdamP
Copyright (c) 2020-present NAVER Corp.
MIT license
AdamP Optimizer Implementation copied from https://github.com/clovaai/AdamP/blob/master/adamp/adamp.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

Loading…
Cancel
Save