Ross Wightman d66819d1f3 Indentation mistake. Fixes #81

2020-02-04 22:56:00 -08:00

..

__init__.py

Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together

2020-01-05 19:58:59 -08:00

auto_augment.py

Update AugMix, JSD, etc comments and references

2020-01-11 12:02:05 -08:00

config.py

…

constants.py

…

dataset.py

Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.

2020-02-01 18:07:32 -08:00

distributed_sampler.py

…

loader.py

Merge pull request #74 from rwightman/augmix-jsd

2020-01-11 12:04:29 -08:00

mixup.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

random_erasing.py

Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together

2020-01-05 19:58:59 -08:00

tf_preprocessing.py

…

transforms_factory.py

Indentation mistake. Fixes #81

2020-02-04 22:56:00 -08: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