Ross Wightman f67a7ee8bd Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct

2022-10-11 15:11:18 -07:00

..

parsers

Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct

2022-10-11 15:11:18 -07:00

__init__.py

Improve image extension handling, add methods to modify / get defaults. Fix #1335 fix #1274.

2022-07-07 14:23:20 -07:00

auto_augment.py

Follow up to #1256, fix interpolation warning in auto_autoaugment as well

2022-06-21 14:56:53 -07:00

config.py

support test_crop_size in data config resolve

2022-07-02 15:17:05 -07:00

constants.py

Add hf hub entries for laion2b clip models, add huggingface_hub dependency, update some setup/reqs, torch >= 1.7

2022-09-16 21:39:05 -07:00

dataset_factory.py

Data improvements. Improve train support for in_chans != 3. Add wds dataset support from bits_and_tpu branch w/ fixes and tweaks. TFDS tweaks.

2022-09-29 16:42:58 -07:00

dataset.py

Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct

2022-10-11 15:11:18 -07:00

distributed_sampler.py

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

loader.py

Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct

2022-10-11 15:11:18 -07:00

mixup.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

random_erasing.py

Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.

2022-09-23 16:08:59 -07:00

real_labels.py

…

tf_preprocessing.py

…

transforms_factory.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

transforms.py

use Image.Resampling namespace for PIL mapping (#1256)

2022-06-12 22:30:57 -07:00