Ross Wightman 2d8b09fe8b Add official pretrained weights to MLP-Mixer, complete model cfgs.

2021-05-05 15:59:40 -07:00

..

data

Enable uniform augmentation magnitude sampling and set AugMix default

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

loss

asymmetric_loss

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

models

Add official pretrained weights to MLP-Mixer, complete model cfgs.

2021-05-05 15:59:40 -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.md with latest, bump version to 0.4.8

2021-04-12 13:15:00 -07:00