replace dist.spawn with dist.launch, test=asr

examples/aishell/asr0/local/train.sh

@@ -20,8 +20,7 @@ if [ ${seed} != 0 ]; then
     export FLAGS_cudnn_deterministic=True
 fi
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --model_type ${model_type} \

examples/aishell/asr1/local/train.sh

@@ -27,9 +27,8 @@ ckpt_name=$2
 
 mkdir -p exp
 
-python3 -u ${BIN_DIR}/train.py \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --seed ${seed} \
---ngpu ${ngpu} \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --profiler-options "${profiler_options}" \

examples/callcenter/asr1/local/train.sh

@@ -21,8 +21,7 @@ if [ ${seed} != 0 ]; then
     export FLAGS_cudnn_deterministic=True
 fi
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --seed ${seed}

examples/librispeech/asr0/local/train.sh

@@ -20,8 +20,7 @@ if [ ${seed} != 0 ]; then
     export FLAGS_cudnn_deterministic=True
 fi
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --model_type ${model_type} \

examples/librispeech/asr1/local/train.sh

@@ -22,8 +22,7 @@ fi
 # export FLAGS_cudnn_exhaustive_search=true
 # export FLAGS_conv_workspace_size_limit=4000
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --seed ${seed}

examples/librispeech/asr2/local/train.sh

@@ -19,9 +19,8 @@ if [ ${seed} != 0 ]; then
     export FLAGS_cudnn_deterministic=True
 fi
 
-python3 -u ${BIN_DIR}/train.py \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --model-name u2_kaldi \
---ngpu ${ngpu} \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --seed ${seed}

examples/timit/asr1/local/train.sh

@@ -19,8 +19,7 @@ if [ ${seed} != 0  ]; then
     export FLAGS_cudnn_deterministic=True
 fi
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --seed ${seed}

examples/tiny/asr0/local/train.sh

@@ -26,8 +26,7 @@ model_type=$3
 
 mkdir -p exp
 
-python3 -u ${BIN_DIR}/train.py \
---ngpu ${ngpu} \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --model_type ${model_type} \

examples/tiny/asr1/local/train.sh

@@ -27,9 +27,8 @@ ckpt_name=$2
 
 mkdir -p exp
 
-python3 -u ${BIN_DIR}/train.py \
+python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
 --seed ${seed} \
---ngpu ${ngpu} \
 --config ${config_path} \
 --output exp/${ckpt_name} \
 --profiler-options "${profiler_options}" \

paddlespeech/s2t/exps/deepspeech2/bin/train.py

@@ -27,9 +27,6 @@ def main_sp(config, args):
 
 
 def main(config, args):
-    if args.ngpu > 1:
-        dist.spawn(main_sp, args=(config, args), nprocs=args.ngpu)
-    else:
     main_sp(config, args)
 
 

paddlespeech/s2t/exps/u2/bin/train.py

@@ -32,9 +32,6 @@ def main_sp(config, args):
 
 
 def main(config, args):
-    if args.ngpu > 1:
-        dist.spawn(main_sp, args=(config, args), nprocs=args.ngpu)
-    else:
     main_sp(config, args)
 
 

paddlespeech/s2t/exps/u2_kaldi/bin/train.py

@@ -36,9 +36,6 @@ def main_sp(config, args):
 
 
 def main(config, args):
-    if args.ngpu > 1:
-        dist.spawn(main_sp, args=(config, args), nprocs=args.ngpu)
-    else:
     main_sp(config, args)
 
 

paddlespeech/s2t/exps/u2_st/bin/train.py

@@ -30,9 +30,6 @@ def main_sp(config, args):
 
 
 def main(config, args):
-    if args.ngpu > 1:
-        dist.spawn(main_sp, args=(config, args), nprocs=args.ngpu)
-    else:
     main_sp(config, args)