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
|
--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 \
|
||||||
|
Loading…
x
Reference in New Issue
Block a user