Ross Wightman 4666cc9aed Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script

2020-01-02 16:22:06 -08:00

..

__init__.py

Adding RandAugment to AutoAugment impl, some tweaks to AA included

2019-11-21 21:14:33 -08:00

auto_augment.py

Update Auto/RandAugment comments, README, more.

2019-11-22 13:24:52 -08:00

config.py

…

constants.py

…

dataset.py

Add basic image folder style dataset to read directly out of tar files, example in validate.py

2019-07-25 10:51:03 -07:00

distributed_sampler.py

…

loader.py

Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script

2020-01-02 16:22:06 -08:00

mixup.py

…

random_erasing.py

Switch RandoErasing back to on GPU normal sampling

2019-12-05 22:35:08 -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

Update Auto/RandAugment comments, README, more.

2019-11-22 13:24:52 -08:00