Ross Wightman b7de82e835 ConViT cleanup, fix torchscript, bit of reformatting, reuse existing layers.

2021-05-21 17:04:23 -07:00

..

data

Enable uniform augmentation magnitude sampling and set AugMix default

2021-04-19 14:21:12 -07:00

loss

…

models

ConViT cleanup, fix torchscript, bit of reformatting, reuse existing layers.

2021-05-21 17:04:23 -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

…

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