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

..

at_datamodule.py

Support AudioSet training with weighted sampler (#1727)

2024-08-22 15:27:25 +08:00

encoder_interface.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00

evaluate.py

Removed unused `k2` dependencies from the AT recipe (#1633)

2024-05-21 18:22:19 +08:00

export-onnx.py

Removed unused `k2` dependencies from the AT recipe (#1633)

2024-05-21 18:22:19 +08:00

export.py

Add CI test for the AudioSet recipe. (#1585)

2024-04-09 17:45:00 +08:00

jit_pretrained.py

Removed unused `k2` dependencies from the AT recipe (#1633)

2024-05-21 18:22:19 +08:00

model.py

Removed unused `k2` dependencies from the AT recipe (#1633)

2024-05-21 18:22:19 +08:00

onnx_pretrained.py

Removed unused `k2` dependencies from the AT recipe (#1633)

2024-05-21 18:22:19 +08:00

optim.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00

pretrained.py

Add CI test for the AudioSet recipe. (#1585)

2024-04-09 17:45:00 +08:00

scaling_converter.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00

scaling.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00

subsampling.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

zipformer.py

Add zipformer recipe for audio tagging (#1421)

2024-04-09 12:06:14 +08:00