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

CTC attention model with reworked Conformer encoder and reworked Transformer decoder (#462)

2022-07-22 15:31:25 +08:00

asr_datamodule.py

CTC attention model with reworked Conformer encoder and reworked Transformer decoder (#462)

2022-07-22 15:31:25 +08:00

attention.py

apply black on all files

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

conformer.py

apply black on all files

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

decode.py

apply black on all files

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

export.py

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

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

label_smoothing.py

CTC attention model with reworked Conformer encoder and reworked Transformer decoder (#462)

2022-07-22 15:31:25 +08:00

optim.py

CTC attention model with reworked Conformer encoder and reworked Transformer decoder (#462)

2022-07-22 15:31:25 +08:00

scaling.py

CTC attention model with reworked Conformer encoder and reworked Transformer decoder (#462)

2022-07-22 15:31:25 +08:00

subsampling.py

Tedlium3 conformer ctc2 (#696)

2022-12-13 16:13:26 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

transformer.py

apply black on all files

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