mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-03 22:24:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
jinzr 2024-02-05 11:48:23 +08:00
parent b07d5472c5
commit 568b7501d1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/aishell/ASR/whisper/train.py
View File

@ -19,7 +19,7 @@
Usage: Usage:
#fine-tuning with deepspeed zero stage 1 #fine-tuning with deepspeed zero stage 1
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_large_v2 \ --exp-dir whisper/exp_large_v2 \
--model-name large-v2 \ --model-name large-v2 \
@ -28,7 +28,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 \
--manifest-dir data/fbank_whisper \ --manifest-dir data/fbank_whisper \
@ -136,7 +136,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