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 670c61b28f
Some cutmix/mixup cleanup/fixes
4 years ago
..
__init__.py Working on CutMix impl as per #8, integrating with Mixup, currently experimenting... 4 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 5 years ago
constants.py Big re-org, working towards making pip/module as 'timm' 5 years ago
dataset.py Working on CutMix impl as per #8, integrating with Mixup, currently experimenting... 4 years ago
distributed_sampler.py Big re-org, working towards making pip/module as 'timm' 5 years ago
loader.py Merge pull request #74 from rwightman/augmix-jsd 5 years ago
mixup.py Some cutmix/mixup cleanup/fixes 4 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 Indentation mistake. Fixes #81 4 years ago