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 0e16d4e9fb
Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.
4 years ago
..
__init__.py Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface. 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
adamw.py
lookahead.py
nadam.py
novograd.py
nvnovograd.py
optim_factory.py Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface. 4 years ago
radam.py
rmsprop_tf.py
sgdp.py