Ross Wightman d7d3538335 Add so400m model size for test, few tweaks.

2025-05-23 18:29:30 -07:00

..

_info

…

readers

…

__init__.py

Add so400m model size for test, few tweaks.

2025-05-23 18:29:30 -07:00

auto_augment.py

…

config.py

…

constants.py

…

dataset_factory.py

remove download argument from torch_kwargs

2025-05-09 07:34:32 -07:00

dataset_info.py

…

dataset.py

…

distributed_sampler.py

…

imagenet_info.py

…

loader.py

…

mixup.py

…

naflex_dataset.py

Add so400m model size for test, few tweaks.

2025-05-23 18:29:30 -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

NaFlex random erasing performance improvements, python loops were slow. Remove subregion mode, not going to be worth it.

2025-05-20 17:03:46 -07:00

naflex_transforms.py

Add so400m model size for test, few tweaks.

2025-05-23 18:29:30 -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

…