Ross Wightman a41de1f666 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

..

parsers

Fix tf.data options setting for newer TF versions

2021-05-15 22:56:12 -07:00

__init__.py

More dataset work including factories and a tensorflow datasets (TFDS) wrapper

2021-01-15 17:26:20 -08:00

auto_augment.py

Add repr to auto_augment and random_erasing impl

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

config.py

Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.

2021-02-23 17:31:42 -08:00

constants.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

dataset_factory.py

Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.

2021-02-23 17:31:42 -08:00

dataset.py

Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.

2021-02-23 17:31:42 -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

Fix a bug with accuracy retrieving from RealLabels

2020-12-04 16:12:50 +01:00

tf_preprocessing.py

Remove some old __future__ imports

2021-01-03 11:24:16 -08:00

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