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.

Hyeongchan Kim a0b2657497 Use `torch.repeat_interleave()` to generate repeated indices faster (#1058) ... * update: use numpy to generate repeated indices faster * update: use torch.repeat_interleave() instead of np.repeat() * refactor: remove unused import, numpy * refactor: torch.range to torch.arange * update: tensor to list before appending the extra samples * update: concatenate the paddings with torch.cat		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