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

..

asr_datamodule.py

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00

decode.py

Add base choice to model_name argument for whisper model. (#1573)

2024-04-02 09:47:38 +08:00

ds_config_zero1.json

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00

label_smoothing.py

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00

optim.py

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00

requirements.txt

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00

train.py

Replace deprecated pytorch methods (#1814)

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

whisper_encoder_forward_monkey_patch.py

Whisper Fine-tuning Recipe on Aishell1 (#1466)

2024-01-27 00:32:30 +08:00