4 Commits

Include renames

Author	SHA1	Message	Date
Peiyuan Zhang	1f423fca06	Fix warning with dataset handling and model loading	2024-10-25 02:31:07 +00:00
Peiyuan Zhang	da9c1ab7c1	Update torchvision imports	2024-10-24 22:11:25 +00:00
LinB203	cea0c041df	update train script ... Former-commit-id: ac16f77d19488cd46130587369cd7a5ea378052b	2024-04-06 13:28:29 +08:00
sennnnn	f3274235d8	Support deepspeed zero2 and zero2_offload for latte training. ... Former-commit-id: c87e95dd357e6fa6d6fb2dcfee781bc49c039e13	2024-03-10 11:27:33 +08:00