14 Commits

Author	SHA1	Message	Date
Xiaoyu Yang	a6c02a4d8c	zipformer BF16 training recipe (#1700) ... Support Zipformer AMP +BF16 training	2024-08-23 09:42:22 +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
Yifan Yang	d65187ec52	Small fix (#1686)	2024-07-11 14:45:35 +08:00
Fangjun Kuang	b88062292b	Typo fixes (#1643)	2024-06-03 16:49:21 +08:00
Zengwei Yao	0df406c5da	Initialize BiasNorm bias with small random values (#1630)	2024-05-20 22:32:02 +08:00
zr_jin	242002e0bd	Strengthened style constraints (#1527)	2024-03-04 23:28:04 +08:00
zr_jin	ef5da4824d	formatted the entire LibriSpeech recipe (#1270) ... * formatted the entire librispeech recipe * minor updates	2023-09-24 17:31:01 +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
Fangjun Kuang	1dbbd7759e	Add tests for subsample.py and fix typos (#1180)	2023-07-25 14:46:18 +08:00
Fangjun Kuang	9009d028a0	Fix ONNX export for the latest non-streaming zipformer. (#1160)	2023-07-03 23:56:51 +08:00
Fangjun Kuang	c3e23ec8d2	Fix logaddexp for ONNX export (#1158)	2023-07-02 10:30:09 +08:00
MicKot	98d89463f6	zipformer2 logaddexp onnx safe (#1157)	2023-06-30 21:16:40 +08:00
danfu	0cb71ad3bc	add updated zipformer onnx export (#1108) ... Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>	2023-06-12 14:02:23 +08:00
Zengwei Yao	f18b539fbc	Add the upgraded Zipformer model (#1058) ... * add the zipformer codes, copied from branch from_dan_scaled_adam_exp1119 * support model export with torch.jit.script * update RESULTS.md * support exporting streaming model with torch.jit.script * add results of streaming models, with some minor changes * update README.md * add CI test * update k2 version in requirements-ci.txt * update pyproject.toml	2023-05-19 16:47:59 +08:00