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/timm/data
Hyeongchan Kim a0b2657497
Use `torch.repeat_interleave()` to generate repeated indices faster (#1058)
3 years ago
..
parsers Matching two bits_and_tpu changes for TFDs wrapper 3 years ago
__init__.py More dataset work including factories and a tensorflow datasets (TFDS) wrapper 4 years ago
auto_augment.py Add repr to auto_augment and random_erasing impl 3 years ago
config.py Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only. 4 years ago
constants.py Big re-org, working towards making pip/module as 'timm' 5 years ago
dataset.py Dataset work 3 years ago
dataset_factory.py Places365 doesn't exist in some still used torchvision version 3 years ago
distributed_sampler.py Use `torch.repeat_interleave()` to generate repeated indices faster (#1058) 3 years ago
loader.py Add worker_seeding arg to allow selecting old vs updated data loader worker seed for (old) experiment repeatability 3 years ago
mixup.py Fix bugs that Mixup does not work device=cpu 3 years ago
random_erasing.py Add repr to auto_augment and random_erasing impl 3 years ago
real_labels.py Fix a bug with accuracy retrieving from RealLabels 4 years ago
tf_preprocessing.py Remove some old __future__ imports 4 years ago
transforms.py Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738. 3 years ago
transforms_factory.py Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738. 3 years ago