Ross Wightman 288682796f Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7

2021-04-01 16:40:12 -07:00

..

__init__.py

Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7

2021-04-01 16:40:12 -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

Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7

2021-04-01 16:40:12 -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