Update train.py

This commit is contained in:
zr_jin 2024-03-07 16:49:33 +08:00 committed by GitHub
parent e96f533b44
commit 21af721ca1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -27,7 +27,7 @@ torchrun --nproc-per-node 8 ./whisper/train.py \
--deepspeed_config ./whisper/ds_config_zero1.json --deepspeed_config ./whisper/ds_config_zero1.json
# fine-tuning with ddp # fine-tuning with ddp
torchrun --nproc-per-node 8 ./whisper/train.py \ torchrun --nproc_per_node 8 ./whisper/train.py \
--max-duration 200 \ --max-duration 200 \
--exp-dir whisper/exp_medium \ --exp-dir whisper/exp_medium \
--base-lr 1e-5 \ --base-lr 1e-5 \