Ross Wightman e846b2cf28 Add 384x384 in12k pretrain and finetune for convnext_nano

2024-12-31 13:16:43 -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 19:24:21 -08:00

loss

…

models

Add 384x384 in12k pretrain and finetune for convnext_nano

2024-12-31 13:16:43 -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

…

__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