Ross Wightman 9d8ccd2ba7 A bit of lars/lamb cleanup, torch.where supports scalars properly now, make lamb grad clipping optional, clean it up a bit		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	A bit of lars/lamb cleanup, torch.where supports scalars properly now, make lamb grad clipping optional, clean it up a bit	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