mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

add multi-node

Browse Source
This commit is contained in:
root 2025-05-18 18:47:22 -07:00
parent 4a29430349
commit 50fc1aba60
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/speech_llm/SPEECH2SPEECH/prepare.sh
View File

@ -295,7 +295,7 @@ if [ $stage -le 16 ] && [ $stop_stage -ge 16 ]; then
done
fi
train_cmd_args="--max-duration 1200 \
train_cmd_args="--max-duration 800 \
--enable-musan False \
--audio-key audio --text-key continuation \
--exp-dir $exp_dir \
@ -318,7 +318,7 @@ if [ $stage -le 16 ] && [ $stop_stage -ge 16 ]; then
# No pretrained model or sampler state dict needed for the first run
fi
torchrun --nproc_per_node $ngpu ./qwen_omni/train.py \
torchrun --nproc_per_node $ngpu --nnodes $SLURM_JOB_NUM_NODES --rdzv_endpoint $MASTER_ADDR:$MASTER_PORT --rdzv_backend c10d --rdzv_id $SLURM_JOBID ./qwen_omni/train.py \
$train_cmd_args
fi