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
Jasha 7c56c718f3
Configure create_optimizer with args.opt_args
4 years ago
..
__init__.py Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support. 4 years ago
adafactor.py Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support. 4 years ago
adahessian.py Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support. 4 years ago
adamp.py Add `adamp` and 'sgdp' optimizers. 4 years ago
adamw.py
lookahead.py Fix some attributions, add copyrights to some file docstrings 4 years ago
nadam.py
novograd.py
nvnovograd.py Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers 5 years ago
optim_factory.py Configure create_optimizer with args.opt_args 4 years ago
radam.py
rmsprop_tf.py Fix some attributions, add copyrights to some file docstrings 4 years ago
sgdp.py Add `adamp` and 'sgdp' optimizers. 4 years ago