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

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

asr_datamodule.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

beam_search.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

decode.py

minor fixes (#1345)

2023-10-27 13:35:43 +08:00

decoder.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

encoder_interface.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

export.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

joiner.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

local

added softlinks to local dir (#1273)

2023-09-26 15:41:39 +08:00

model.py

Replace deprecated pytorch methods (#1814)

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

my_profile.py

Add CI test to cover zipformer/train.py (#1424)

2023-12-23 00:38:36 +08:00

optim.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

pretrained.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

scaling_converter.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

scaling.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

subsampling.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

zipformer.py

Zipformer for TedLium (#1125)

2023-06-28 16:43:49 +08:00