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

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

asr_datamodule.py

Fix buffer size of DynamicBucketingSampler (#1468)

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

beam_search.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

conformer.py

Update conformer.py for aishell4 (#484)

2022-07-20 21:32:53 +08:00

decode.py

Fix style issues. (#937)

2023-03-08 22:56:04 +08:00

decoder.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

encoder_interface.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

export.py

fixed a CI test for wenetspeech (#1476)

2024-01-27 06:41:56 +08:00

joiner.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

local

Migrate zipformer model to other Chinese datasets (#1216)

2023-10-24 16:24:46 +08:00

model.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

optim.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

pretrained.py

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

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

scaling.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

test_model.py

[Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399)

2022-06-14 22:19:05 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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