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.
Ross Wightman
d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more
|
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
|
5 years ago |
constants.py
|
Big re-org, working towards making pip/module as 'timm'
|
5 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'
|
5 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 |