You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/data
Ross Wightman d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
5 years ago
..
__init__.py Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together 5 years ago
auto_augment.py Update AugMix, JSD, etc comments and references 5 years ago
config.py A number of tweaks to arguments, epoch handling, config 6 years ago
constants.py Big re-org, working towards making pip/module as 'timm' 6 years ago
dataset.py Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together 5 years ago
distributed_sampler.py Big re-org, working towards making pip/module as 'timm' 6 years ago
loader.py Merge pull request #74 from rwightman/augmix-jsd 5 years ago
mixup.py Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl 5 years ago
random_erasing.py Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together 5 years ago
tf_preprocessing.py Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 5 years ago
transforms.py Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl 5 years ago
transforms_factory.py Update AugMix, JSD, etc comments and references 5 years ago