icefall/egs/librispeech/ASR/pruned_transducer_stateless3
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
..
2023-04-03 16:20:29 +08:00
2023-04-21 18:09:41 +08:00
2023-09-21 21:16:14 +08:00
2022-12-02 10:53:06 +08:00