Ross Wightman 13e0f3a4a3 Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model.		2025-04-08 20:47:57 -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.py	…
dataset_factory.py	…
dataset_info.py	…
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	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	Type fixes, remove old comments	2025-04-07 21:35:03 -07:00
random_erasing.py	…
real_labels.py	…
tf_preprocessing.py	…
transforms.py	…
transforms_factory.py	Initial NaFlex ViT model and training support	2025-04-07 21:27:10 -07:00