Ross Wightman fb94350896 Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss

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

..

parsers

Fix tf.data options setting for newer TF versions

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

__init__.py

…

auto_augment.py

Add mmax config key to auto_augment for increasing upper bound of RandAugment magnitude beyond 10. Make AugMix uniform sampling default not override config setting.

2021-08-12 15:39:05 -07:00

config.py

…

constants.py

…

dataset_factory.py

…

dataset.py

…

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

Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss

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

mixup.py

…

random_erasing.py

…

real_labels.py

…

tf_preprocessing.py

…

transforms_factory.py

Support for huggingface hub via create_model and default_cfgs.

2021-03-16 22:48:26 -07:00

transforms.py

Support for huggingface hub via create_model and default_cfgs.

2021-03-16 22:48:26 -07:00