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

..

.gitignore

Add tests for subsample.py and fix typos (#1180)

2023-07-25 14:46:18 +08:00

asr_datamodule.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

attention_decoder.py

A LibriTTS recipe on both ASR & Neural Codec Tasks (#1746)

2024-10-21 11:30:14 +08:00

beam_search.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

ctc_decode.py

fixed formatting issue (#1791)

2024-10-30 21:14:12 +08:00

decode_gigaspeech.py

Fine-tune recipe for Zipformer (#1484)

2024-02-06 18:25:43 +08:00

decode_stream.py

Fix decode_stream.py (#1208)

2023-08-09 09:40:58 +08:00

decode.py

split save_results() -> save_asr_output() + save_wer_results() (#1712)

2024-08-13 23:02:14 +08:00

decoder.py

minor fixes (#1332)

2023-10-24 08:17:17 +08:00

encoder_interface.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

export-onnx-ctc.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

export-onnx-streaming-ctc.py

Add streaming HLG decoding for zipformer CTC. (#1557)

2024-03-18 20:11:47 +08:00

export-onnx-streaming.py

A LibriTTS recipe on both ASR & Neural Codec Tasks (#1746)

2024-10-21 11:30:14 +08:00

export-onnx.py

A LibriTTS recipe on both ASR & Neural Codec Tasks (#1746)

2024-10-21 11:30:14 +08:00

export.py

Support CTC/AED option for Zipformer recipe (#1389)

2024-07-05 20:19:18 +08:00

finetune.py

Replace deprecated pytorch methods (#1814)

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

generate_averaged_model.py

zipformer wenetspeech (#1130)

2023-06-26 09:33:18 +08:00

jit_pretrained_ctc.py

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

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

jit_pretrained_streaming.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

jit_pretrained.py

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

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

joiner.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

label_smoothing.py

Support CTC/AED option for Zipformer recipe (#1389)

2024-07-05 20:19:18 +08:00

model.py

Replace deprecated pytorch methods (#1814)

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

my_profile.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

onnx_check.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

onnx_decode.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

onnx_pretrained_ctc_H.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

onnx_pretrained_ctc_HL.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

onnx_pretrained_ctc_HLG_streaming.py

Add streaming HLG decoding for zipformer CTC. (#1557)

2024-03-18 20:11:47 +08:00

onnx_pretrained_ctc_HLG.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

onnx_pretrained_ctc.py

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

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

onnx_pretrained-streaming-ctc.py

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

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

onnx_pretrained-streaming.py

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

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

onnx_pretrained.py

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

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

optim.py

Avoid using lr from checkpoint. (#1781)

2024-10-28 00:59:04 +08:00

pretrained_ctc.py

Support CTC/AED option for Zipformer recipe (#1389)

2024-07-05 20:19:18 +08:00

pretrained.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

scaling_converter.py

Fix typos, remove unused packages, normalize comments (#1678)

2024-07-04 14:19:45 +08:00

scaling.py

Replace deprecated pytorch methods (#1814)

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

streaming_beam_search.py

zipformer wenetspeech (#1130)

2023-06-26 09:33:18 +08:00

streaming_decode.py

split save_results() -> save_asr_output() + save_wer_results() (#1712)

2024-08-13 23:02:14 +08:00

subsampling.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

test_scaling.py

Add tests for subsample.py and fix typos (#1180)

2023-07-25 14:46:18 +08:00

test_subsampling.py

Add tests for subsample.py and fix typos (#1180)

2023-07-25 14:46:18 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

zipformer.py

Replace deprecated pytorch methods (#1814)

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