From 362a6de3c850079ed2ae6f94181f2a51c0596527 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 15:22:13 +0800
Subject: [PATCH] update
---
examples/aishell/conformer/run.sh | 11 ++++++++---
1 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/examples/aishell/conformer/run.sh b/examples/aishell/conformer/run.sh
index 28e6f90..947cc02 100755
--- a/examples/aishell/conformer/run.sh
+++ b/examples/aishell/conformer/run.sh
@@ -109,9 +109,14 @@
echo "log_file: ${log_file}"
gpu_num=$(echo CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
- torchrun \
- --nnodes 1 \
- --nproc_per_node ${gpu_num} \
+# 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} \
../../../funasr/bin/train.py \
--config-path "${workspace}/conf" \
--config-name "${config}" \
--
Gitblit v1.9.1