Ross Wightman cd23f55397 Fix mixed prec issues with new mixup code

2020-08-11 12:17:43 -07:00

..

__init__.py

Merge branch 'master' into cutmix. Fixup a few issues.

2020-08-07 15:59:52 -07:00

auto_augment.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

config.py

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

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

constants.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

dataset.py

Merge branch 'master' into cutmix. Fixup a few issues.

2020-08-07 15:59:52 -07:00

distributed_sampler.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

loader.py

Add more augmentation arguments, including a no_aug disable flag. Fix #209

2020-08-05 13:16:44 -07:00

mixup.py

Fix mixed prec issues with new mixup code

2020-08-11 12:17:43 -07:00

random_erasing.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

real_labels.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

tf_preprocessing.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

transforms_factory.py

More documentation updates, fix a typo

2020-08-05 15:59:31 -07:00

transforms.py

Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl

2019-12-20 23:04:11 -08:00