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

..

__init__.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

asr_datamodule.py

update speechio whisper ft results (#1605)

2024-04-30 11:49:20 +08:00

beam_search.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

conformer.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

decode_aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

decode.py

fix typo (#1324)

2023-10-19 22:54:43 +08:00

decoder.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

encoder_interface.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

export-onnx.py

fixed a CI test for wenetspeech (#1476)

2024-01-27 06:41:56 +08:00

export.py

Fix torchscript export to use tokens.txt instead of lang_dir (#1475)

2024-01-26 19:18:33 +08:00

finetune.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00

jit_pretrained.py

Use high_freq -400 in computing fbank features. (#1447)

2024-01-04 13:59:32 +08:00

joiner.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

lstmp.py

Support exporting LSTM with projection to ONNX (#621)

2022-10-18 11:25:31 +08:00

model.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

onnx_check.py

Fix CI tests (#1266)

2023-09-21 21:16:14 +08:00

onnx_pretrained.py

add onnx export for stateless2 (#1086)

2023-05-23 16:11:00 +08:00

optim.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

pretrained.py

Use high_freq -400 in computing fbank features. (#1447)

2024-01-04 13:59:32 +08:00

scaling_converter.py

Support exporting to ONNX for the wenetspeech recipe (#615)

2022-10-13 15:17:20 +08:00

scaling.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

train.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00