diff --git a/examples/speechcommand_v1/s0/run.sh b/examples/speechcommand_v1/s0/run.sh index b18a856..235396c 100755 --- a/examples/speechcommand_v1/s0/run.sh +++ b/examples/speechcommand_v1/s0/run.sh @@ -4,8 +4,6 @@ . ./path.sh -export CUDA_VISIBLE_DEVICES="0" - stage=-1 stop_stage=4 num_keywords=11 @@ -13,7 +11,7 @@ num_keywords=11 config=conf/mdtc.yaml norm_mean=false norm_var=false -gpu_id=0 +gpus="0" checkpoint= dir=exp/mdtc @@ -22,7 +20,7 @@ num_average=10 score_checkpoint=$dir/avg_${num_average}.pt # your data dir -download_dir=/mnt/mnt-data-3/jingyong.hou/data +download_dir=./data/local speech_command_dir=$download_dir/speech_commands_v1 . tools/parse_options.sh || exit 1; @@ -68,7 +66,9 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then cmvn_opts= $norm_mean && cmvn_opts="--cmvn_file data/train/global_cmvn" $norm_var && cmvn_opts="$cmvn_opts --norm_var" - python kws/bin/train.py --gpu $gpu_id \ + num_gpus=$(echo $gpus | awk -F ',' '{print NF}') + torchrun --standalone --nnodes=1 --nproc_per_node=$num_gpus \ + kws/bin/train.py --gpus $gpus \ --config $config \ --train_data data/train/data.list \ --cv_data data/valid/data.list \