Ross Wightman 0e16d4e9fb Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.

2021-02-23 15:38:12 -08:00

..

__init__.py

Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.

2021-02-23 15:38:12 -08:00

adafactor.py

Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.

2020-10-09 17:24:43 -07:00

adahessian.py

Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.

2020-10-09 17:24:43 -07:00

adamp.py

Add adamp and 'sgdp' optimizers.

2020-07-25 15:33:20 -07:00

adamw.py

…

lookahead.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

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.

2021-02-23 15:38:12 -08:00

radam.py

…

rmsprop_tf.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

sgdp.py

Add adamp and 'sgdp' optimizers.

2020-07-25 15:33:20 -07:00