Ross Wightman 2a84d68d02 Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe

2025-01-18 15:51:57 -08:00

..

data

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

layers

Add missing imports

2025-01-11 15:13:16 +01:00

loss

…

models

Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe

2025-01-18 15:51:57 -08:00

optim

…

scheduler

…

utils

Update checkpoint save to fix old hard-link + fuse issue I ran into again... fix #340

2025-01-08 15:36:58 -08:00

__init__.py

…

py.typed

…

version.py

Update version.py

2025-01-16 11:23:17 -08:00