Ross Wightman 19090ea966 Need to init momentum with correct dtype

2024-11-12 20:49:01 -08:00

..

data

Add trust_remote_code argument to ReaderHfds

2024-11-08 08:16:36 -08:00

layers

…

loss

…

models

fix pos embed dynamic resampling for eva

2024-11-06 16:03:27 -08:00

optim

Need to init momentum with correct dtype

2024-11-12 20:49:01 -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

Merge branch 'MengqingCao-npu_support' into device_amp_cleanup

2024-10-18 14:50:44 -07:00

__init__.py

…

version.py

Update version.py

2024-10-19 09:56:17 -07:00