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

GigaSpeech RNN-T experiments (#318)

2022-05-13 11:03:26 +08:00

asr_datamodule.py

Fix buffer size of DynamicBucketingSampler (#1468)

2024-01-21 02:10:42 +08:00

beam_search.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

compute_ppl.py

Add compute_ppl.py and ngram_entropy_pruning.py (#1013)

2023-04-20 12:27:43 +08:00

conformer.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

decode.py

Fix CI test for gigaspeech (#1787)

2024-10-29 15:50:49 +08:00

decoder.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

encoder_interface.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

export.py

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

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

gigaspeech_scoring.py

GigaSpeech RNN-T experiments (#318)

2022-05-13 11:03:26 +08:00

joiner.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

lstmp.py

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

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

model.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

optim.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

scaling_converter.py

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

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

scaling.py

Update decoding script for gigaspeech and remove duplicate files. (#361)

2022-05-13 13:03:16 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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