Ross Wightman 3dc90ed7a7 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

..

_info

…

readers

Fix typos

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

__init__.py

Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model.

2025-04-08 20:47:57 -07:00

auto_augment.py

Fix typos

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

config.py

…

constants.py

…

dataset_factory.py

…

dataset_info.py

…

dataset.py

…

distributed_sampler.py

…

imagenet_info.py

Fix typos

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

loader.py

…

mixup.py

…

naflex_dataset.py

Initial NaFlex ViT model and training support

2025-04-07 21:27:10 -07:00

naflex_loader.py

Type fixes, remove old comments

2025-04-07 21:35:03 -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

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

transforms.py

…