Ross Wightman d9a6a9d0af

Merge pull request #74 from rwightman/augmix-jsd ...

AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more

2020-01-11 12:04:29 -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

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

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

dataset.py

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

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

distributed_sampler.py

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

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

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

Update AugMix, JSD, etc comments and references

2020-01-11 12:02:05 -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