Ross Wightman bfc72f75d3 Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.

2021-05-24 21:13:26 -07:00

..

data

Fix tf.data options setting for newer TF versions

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

loss

…

models

Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.

2021-05-24 21:13:26 -07:00

optim

…

scheduler

…

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