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 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		6 years ago
..
__init__.py	Adding RandAugment to AutoAugment impl, some tweaks to AA included	6 years ago
auto_augment.py	Update Auto/RandAugment comments, README, more.	6 years ago
config.py	…
constants.py	…
dataset.py	…
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	6 years ago
mixup.py	…
random_erasing.py	Switch RandoErasing back to on GPU normal sampling	6 years ago
tf_preprocessing.py	…
transforms.py	Update Auto/RandAugment comments, README, more.	6 years ago