4 Commits

Author	SHA1	Message	Date
Li Peng	2d9825aa29	Replace deprecated pytorch methods ... - torch.cuda.amp.GradScaler(...) => torch.amp.GradScaler("cuda", ...) - torch.cuda.amp.autocast(...) => torch.amp.autocast("cuda", ...)	2024-11-27 17:43:17 +08:00
Zengwei Yao	334beed2af	fix usages of returned losses after adding attention-decoder in zipformer (#1689)	2024-07-12 16:50:58 +08:00
Teo Wen Shen	19048e155b	Cast grad_scale in whiten to float (#1663) ... * cast grad_scale in whiten to float * fix cast in zipformer_lora	2024-07-11 15:12:30 +08:00
Xiaoyu Yang	2dfd5dbf8b	Add LoRA for Zipformer (#1540)	2024-03-15 17:19:23 +08:00