mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 02:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source 
minor updates
This commit is contained in:
zr_jin 2024-03-07 16:45:48 +08:00 committed by GitHub
parent 4cca65ac98
commit 211ce4c5fd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/wenetspeech/ASR/whisper/train.py
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 \
@ -134,7 +134,7 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--exp-dir", "--exp-dir",
type=str, type=str,
default="pruned_transducer_stateless7/exp", default="whisper/exp",
help="""The experiment dir. help="""The experiment dir.
It specifies the directory where all training related It specifies the directory where all training related
files, e.g., checkpoints, log, etc, are saved files, e.g., checkpoints, log, etc, are saved