mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Update train.py
This commit is contained in:
parent
e96f533b44
commit
21af721ca1
@ -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 \
|
||||
|
Loading…
x
Reference in New Issue
Block a user