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

Start adding some files..

2022-03-16 22:14:30 +08:00

asr_datamodule.py

Start adding some files..

2022-03-16 22:14:30 +08:00

beam_search.py

Recipes for open vocabulary keyword spotting (#1428)

2024-02-22 15:31:20 +08:00

conformer.py

Fix padding issues (#1303)

2023-10-11 16:58:00 +08:00

decode_stream.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

decode.py

add option of using full attention for streaming model decoding (#975)

2023-03-30 14:30:13 +08:00

decoder.py

Fix padding_idx (#942)

2023-03-10 14:37:28 +08:00

encoder_interface.py

Add remaining files in pruned_transducer_stateless2

2022-03-17 11:17:52 +08:00

export.py

Fix torch.jit.script() export for pruned_transducer_stateless2 (#1410)

2023-12-10 11:38:39 +08:00

joiner.py

Simplify ONNX export (#881)

2023-02-07 15:01:59 +08:00

lstmp.py

Fix torch.jit.script() export for pruned_transducer_stateless2 (#1410)

2023-12-10 11:38:39 +08:00

model.py

Replace deprecated pytorch methods (#1814)

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

optim.py

formatted the entire LibriSpeech recipe (#1270)

2023-09-24 17:31: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

Fix torch.jit.script() export for pruned_transducer_stateless2 (#1410)

2023-12-10 11:38:39 +08:00

scaling.py

formatted the entire LibriSpeech recipe (#1270)

2023-09-24 17:31:01 +08:00

streaming_beam_search.py

apply black on all files

2022-11-17 09:42:17 -05:00

streaming_decode.py

streaming_decode.py, relax the audio range from [-1,+1] to [-10,+10] (#1448)

2024-01-05 10:21:27 +08:00

test_model.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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