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.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
transforms_factory.py	Update AugMix, JSD, etc comments and references	2020-01-11 12:02:05 -08:00