Ross Wightman a4d8fea61e Add model based wd skip support. Improve cross version compat of optimizer factory. Fix #247		2020-10-13 12:49:47 -07:00
..
__init__.py	Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.	2020-10-09 17:24:43 -07: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	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.	2019-08-28 00:14:10 -07:00
lookahead.py	Fix some attributions, add copyrights to some file docstrings	2020-07-27 13:44:56 -07:00
nadam.py	Big re-org, working towards making pip/module as 'timm'	2019-06-19 17:20:51 -07:00
novograd.py	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.	2019-08-28 00:14:10 -07:00
nvnovograd.py	Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers	2019-08-29 15:21:38 -07:00
optim_factory.py	Add model based wd skip support. Improve cross version compat of optimizer factory. Fix #247	2020-10-13 12:49:47 -07:00
radam.py	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.	2019-08-28 00:14:10 -07:00
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