Ross Wightman 9ec3210c2d More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently).

2021-11-10 15:52:09 -08:00

..

parsers

More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently).

2021-11-10 15:52:09 -08:00

__init__.py

…

auto_augment.py

Add repr to auto_augment and random_erasing impl

2021-10-28 17:33:36 -07:00

config.py

…

constants.py

…

dataset_factory.py

Dataset work

2021-11-09 22:34:15 -08:00

dataset.py

Dataset work

2021-11-09 22:34:15 -08:00

distributed_sampler.py

Add RepeatAugSampler as per DeiT RASampler impl, showing promise for current (distributed) training experiments.

2021-09-01 17:40:53 -07:00

loader.py

Add worker_seeding arg to allow selecting old vs updated data loader worker seed for (old) experiment repeatability

2021-09-28 16:37:45 -07:00

mixup.py

Fix bugs that Mixup does not work device=cpu

2021-10-12 23:51:46 +09:00

random_erasing.py

Add repr to auto_augment and random_erasing impl

2021-10-28 17:33:36 -07:00

real_labels.py

…

tf_preprocessing.py

…

transforms_factory.py

Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738.

2021-10-28 17:35:01 -07:00

transforms.py

Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738.

2021-10-28 17:35:01 -07:00