You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/optim
Ross Wightman 9541f4963b
One more scalar -> tensor fix for lamb optimizer
3 years ago
..
__init__.py A few optimizer comments, dead import, missing import 3 years ago
adabelief.py Remove experiment from lamb impl 3 years ago
adafactor.py Optimizer improvements, additions, cleanup 3 years ago
adahessian.py Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support. 4 years ago
adamp.py Optimizer improvements, additions, cleanup 3 years ago
adamw.py Optimizer improvements, additions, cleanup 3 years ago
lamb.py One more scalar -> tensor fix for lamb optimizer 3 years ago
lookahead.py Optimizer improvements, additions, cleanup 3 years ago
madgrad.py add madgradw optimizer 3 years ago
nadam.py Optimizer improvements, additions, cleanup 3 years ago
nvnovograd.py Optimizer improvements, additions, cleanup 3 years ago
optim_factory.py add madgradw optimizer 3 years ago
radam.py Optimizer improvements, additions, cleanup 3 years ago
rmsprop_tf.py Optimizer improvements, additions, cleanup 3 years ago
sgdp.py Optimizer improvements, additions, cleanup 3 years ago