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

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

asr_datamodule.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

beam_search.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

decode.py

Remove simulate streaming from stateless8 (#985)

2023-04-04 11:04:00 +08:00

decoder.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

encoder_interface.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

export-onnx.py

add export-onnx.py for stateless8 (#1302)

2023-10-11 12:20:12 +08:00

export.py

Use tokens.txt to replace bpe.model (#1162)

2023-08-12 16:53:59 +08:00

gigaspeech.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

jit_pretrained.py

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

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

joiner.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

librispeech.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

model.py

Replace deprecated pytorch methods (#1814)

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

optim.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

pretrained.py

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

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

scaling_converter.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

scaling.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

zipformer.py

Add zipformer from Dan using multi-dataset setup (#675)

2022-11-15 16:56:05 +08:00