Antoine Broyelle 78fa0772cc Leverage python hierachical logger ...

with this update one can tune the kind of logs generated by timm but
training and inference traces are unchanged

2020-06-09 18:28:48 +01: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

Leverage python hierachical logger

2020-06-09 18:28:48 +01:00

constants.py

…

dataset.py

Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.

2020-02-01 18:07:32 -08:00

distributed_sampler.py

…

loader.py

added MultiEpochsDataLoader

2020-05-05 14:47:02 +03: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

Indentation mistake. Fixes #81

2020-02-04 22:56:00 -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