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		5 years ago
..
__init__.py	…
auto_augment.py	…
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	5 years ago
mixup.py	…
random_erasing.py	…
tf_preprocessing.py	…
transforms.py	…