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.
pytorch-image-models/data
Ross Wightman 5180f94c7e
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support
6 years ago
..
__init__.py Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 6 years ago
dataset.py Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 6 years ago
random_erasing.py Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 6 years ago
transforms.py Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 6 years ago
utils.py Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 6 years ago