Ross Wightman 11ae795e99 Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel

2021-05-25 10:15:32 -07:00

..

data

Fix tf.data options setting for newer TF versions

2021-05-15 22:56:12 -07:00

loss

asymmetric_loss

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

models

Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel

2021-05-25 10:15:32 -07:00

optim

Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like

2021-04-12 08:25:31 -07:00

scheduler

Merge branch 'logger' into features. Change 'logger' to '_logger'.

2020-07-27 18:00:46 -07:00

utils

Merge pull request #550 from amaarora/wandb

2021-04-15 09:26:35 -07:00

__init__.py

Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models

2021-04-12 11:54:22 -07:00

version.py

Update README, tweak fine-tune effv2 model names.

2021-05-14 16:37:43 -07:00