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

A number of tweaks to arguments, epoch handling, config

2019-06-28 13:49:20 -07:00

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

Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl

2019-07-25 11:42:01 -07:00

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