Skip to content

Gunale0926/Grams

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Grams: Gradient Descent with Adaptive Momentum Scaling

arXiv PyPI - Version

image

Install

pip install grams-pytorch

How to use

Import:

from grams import Grams

Instantiate:

optimizer = Grams(lr=1e-3, weight_decay=0.0)