From 6c8e8397424b02b6df3248824d24b9f993536f90 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 15:53:06 +0800
Subject: [PATCH] update

---
 examples/aishell/conformer/run.sh |   11 +++--------
 1 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/examples/aishell/conformer/run.sh b/examples/aishell/conformer/run.sh
index 947cc02..28e6f90 100755
--- a/examples/aishell/conformer/run.sh
+++ b/examples/aishell/conformer/run.sh
@@ -109,14 +109,9 @@
   echo "log_file: ${log_file}"
 
   gpu_num=$(echo CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
-#  torchrun \
-#  --nnodes 1 \
-#  --nproc_per_node ${gpu_num}
-  cmd="python"
-  if [ ${gpu_num} -gt 1  ];then
-    cmd="torchrun --nnodes 1 --nproc_per_node ${gpu_num}"
-  fi
-  ${cmd} \
+  torchrun \
+  --nnodes 1 \
+  --nproc_per_node ${gpu_num} \
   ../../../funasr/bin/train.py \
   --config-path "${workspace}/conf" \
   --config-name "${config}" \

--
Gitblit v1.9.1