Ross Wightman 7624389fc9 Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

..

_info

Add mini imagenet info files

2024-11-25 10:53:28 -08:00

readers

Fix typos

2025-01-19 13:39:40 -08:00

__init__.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

auto_augment.py

Fix typos

2025-01-19 13:39:40 -08:00

config.py

…

constants.py

…

dataset_factory.py

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -08:00

dataset_info.py

…

dataset.py

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -08:00

distributed_sampler.py

…

imagenet_info.py

Fix typos

2025-01-19 13:39:40 -08:00

loader.py

Add half-precision (bfloat16, float16) support to train & validate scripts. Should push dtype handling into model factory / pretrained load at some point...

2025-01-07 10:25:14 -08:00

mixup.py

…

naflex_dataset.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

naflex_loader.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

naflex_mixup.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

naflex_random_erasing.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

naflex_transforms.py

Add naflex loader support to validate.py, fix bug in naflex pos embed add, classic vit weight loading for naflex model

2025-04-25 16:00:54 -07:00

random_erasing.py

…

real_labels.py

…

tf_preprocessing.py

…

transforms_factory.py

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

transforms.py

…