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

A number of tweaks to arguments, epoch handling, config

2019-06-28 13:49:20 -07:00

constants.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

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

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

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

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

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