Ross Wightman 736f209e7d Update vision transformers to be compatible with official code. Port official ViT weights from jax impl.

2020-10-26 18:42:11 -07:00

..

data

…

loss

asymmetric_loss

2020-10-16 17:12:28 +03:00

models

Update vision transformers to be compatible with official code. Port official ViT weights from jax impl.

2020-10-26 18:42:11 -07:00

optim

Fix optimizer factory regressin for optimizers like sgd/momentum that don't have an eps arg

2020-10-22 15:59:47 -07:00

scheduler

…

utils

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

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

__init__.py

…

version.py

…