11 Commits

Author	SHA1	Message	Date
Fangjun Kuang	63563d16d3	Fix setting joiner dim (#2027) ... Fixes incorrect computation of encoder_dim when encoder_dim is a comma-separated list of integers by ensuring numeric (not lexicographic) max is used. Fixes #2018 - Replace int(max(params.encoder_dim.split(","))) (lexicographic max on strings) with max(_to_int_tuple(params.encoder_dim)) (numeric max). - Apply the fix consistently across all affected training scripts.	2025-09-19 09:42:41 +08:00
Teo Wen Shen	da87e7fc99	add weights_only=False to torch.load (#1984)	2025-07-10 15:27:08 +08:00
Fangjun Kuang	d4d4f281ec	Revert "Replace deprecated pytorch methods (#1814)" (#1841) ... This reverts commit 3e4da5f78160d3dba3bdf97968bd7ceb8c11631f.	2024-12-18 16:49:57 +08:00
Li Peng	3e4da5f781	Replace deprecated pytorch methods (#1814) ... * Replace deprecated pytorch methods - torch.cuda.amp.GradScaler(...) => torch.amp.GradScaler("cuda", ...) - torch.cuda.amp.autocast(...) => torch.amp.autocast("cuda", ...) * Replace `with autocast(...)` with `with autocast("cuda", ...)` Co-authored-by: Li Peng <lipeng@unisound.ai>	2024-12-16 10:24:16 +08:00
zr_jin	eb132da00d	additional instruction for the grad_scale is too small error (#1550)	2024-03-14 11:33:49 +08:00
zr_jin	242002e0bd	Strengthened style constraints (#1527)	2024-03-04 23:28:04 +08:00
Fangjun Kuang	79a42148db	Add CI test to cover zipformer/train.py (#1424)	2023-12-23 00:38:36 +08:00
zr_jin	ea78b32857	minor fixes (#1345)	2023-10-27 13:35:43 +08:00
zr_jin	1b565dd251	added softlinks to local dir (#1273)	2023-09-26 15:41:39 +08:00
zr_jin	ef658d691e	fixes for init value of diagnostics.TensorDiagnosticOptions (#1269) ... * fixes for `diagnostics` Replace `2 ** 22` with `512` as the default value of `diagnostics.TensorDiagnosticOptions` also black formatted some scripts * fixed formatting issues	2023-09-24 17:06:47 +08:00
Desh Raj	9c2172c1c4	Zipformer for TedLium (#1125) ... * initial commit for zipformer tedlium * fix unk decoding * add pretrained model and logs * update for new AsrModel * add option for choosing rnnt type * add results with modified rnnt	2023-06-28 16:43:49 +08:00