5 Commits

Author	SHA1	Message	Date
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
Dadoou	6cbddaa8e3	Add base choice to model_name argument for whisper model. (#1573) ... Co-authored-by: dadoou <dadoou@yandex.com>	2024-04-02 09:47:38 +08:00
zr_jin	9b0eae3b4a	fixes for init value of diagnostics.TensorDiagnosticOptions (#1555)	2024-03-18 17:14:29 +08:00
zr_jin	a813186f64	minor fix for docstr and default param. (#1490) ... * Update train.py and README.md	2024-02-05 12:47:52 +08:00
Yuekai Zhang	1c30847947	Whisper Fine-tuning Recipe on Aishell1 (#1466) ... * add decode seamlessm4t * add requirements * add decoding with avg model * add token files * add custom tokenizer * support deepspeed to finetune large model * support large-v3 * add model saving * using monkey patch to replace models * add manifest dir option	2024-01-27 00:32:30 +08:00