11 Commits

Author	SHA1	Message	Date
Teo Wen Shen	da87e7fc99	add weights_only=False to torch.load (#1984)	2025-07-10 15:27:08 +08:00
Fangjun Kuang	fba5e67d5e	Fix CI tests. (#1974) ... - Introduce unified AMP helpers (create_grad_scaler, torch_autocast) to handle deprecations in PyTorch ≥2.3.0 - Replace direct uses of torch.cuda.amp.GradScaler and torch.cuda.amp.autocast with the new utilities across all training and inference scripts - Update all torch.load calls to include weights_only=False for compatibility with newer PyTorch versions	2025-07-01 13:47:55 +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	bf2f94346c	Enabling char_level and compute_CER for aishell recipe (#1554) ... * init fix Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>	2024-03-18 11:57:47 +08:00
zr_jin	eb132da00d	additional instruction for the grad_scale is too small error (#1550)	2024-03-14 11:33:49 +08:00
Fangjun Kuang	8136ad775b	Use high_freq -400 in computing fbank features. (#1447) ... See also https://github.com/k2-fsa/sherpa-onnx/issues/514	2024-01-04 13:59:32 +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
Wei Kang	219bba1310	zipformer wenetspeech (#1130) ... * copy files * update train.py * small fixes * Add decode.py * Fix dataloader in decode.py * add blank penalty * Add blank-penalty to other decoding method * Minor fixes * add zipformer2 recipe * Minor fixes * Remove pruned7 * export and test models * Replace bpe with tokens in export.py and pretrain.py * Minor fixes * Minor fixes * Minor fixes * Fix export * Update results * Fix zipformer-ctc * Fix ci * Fix ci * Fix CI * Fix CI --------- Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>	2023-06-26 09:33:18 +08:00
Fangjun Kuang	7b0afbdc16	Remove cur_batch_idx (#1102)	2023-05-30 14:49:54 +08:00
Wei Kang	80156dda09	Training with byte level BPE (AIShell) (#986) ... * copy files from zipformer librispeech * Add byte bpe training for aishell * compile LG graph * Support LG decoding * Minor fixes * black * Minor fixes * export & fix pretrain.py * fix black * Update RESULTS.md * Fix export.py	2023-05-04 19:16:17 +08:00