From 21af721ca141a151e4937e8e1757a7a7bde00bd7 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Thu, 7 Mar 2024 16:49:33 +0800 Subject: [PATCH] Update train.py --- egs/multi_zh-hans/ASR/whisper/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/multi_zh-hans/ASR/whisper/train.py b/egs/multi_zh-hans/ASR/whisper/train.py index 19bd188c5..11a22eec1 100644 --- a/egs/multi_zh-hans/ASR/whisper/train.py +++ b/egs/multi_zh-hans/ASR/whisper/train.py @@ -27,7 +27,7 @@ torchrun --nproc-per-node 8 ./whisper/train.py \ --deepspeed_config ./whisper/ds_config_zero1.json # fine-tuning with ddp -torchrun --nproc-per-node 8 ./whisper/train.py \ +torchrun --nproc_per_node 8 ./whisper/train.py \ --max-duration 200 \ --exp-dir whisper/exp_medium \ --base-lr 1e-5 \