Ross Wightman eb84e4b571 Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change.

2024-12-30 17:06:34 -08:00

..

data

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

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

layers

Add SimpleNorm to create_norm factory

2024-12-30 14:22:42 -08:00

loss

…

models

Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change.

2024-12-30 17:06:34 -08:00

optim

Add caution to Adan. Add decouple decay option to LAMB.

2024-12-05 13:50:30 -08:00

scheduler

Extend train epoch schedule by warmup_epochs if warmup_prefix enable, allows schedule to reach end w/ prefix enabledy

2024-11-08 11:01:11 -08:00

utils

Tweak a few docstrings

2024-11-13 10:12:31 -08:00

__init__.py

…

py.typed

Add py.typed file as recommended by PEP 561

2024-11-14 11:26:00 -08:00

version.py

Back to dev version

2024-12-06 10:44:41 -08:00