From bc19499b48a161d637309b6cb6fefb215a5bfb9f Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 16:22:53 +0800
Subject: [PATCH] update
---
examples/aishell/transformer/run.sh | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/examples/aishell/transformer/run.sh b/examples/aishell/transformer/run.sh
index 276152b..90955f2 100755
--- a/examples/aishell/transformer/run.sh
+++ b/examples/aishell/transformer/run.sh
@@ -108,7 +108,7 @@
log_file="${exp_dir}/exp/${model_dir}/train.log.txt"
echo "log_file: ${log_file}"
- gpu_num=$(echo CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
+ gpu_num=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
torchrun \
--nnodes 1 \
--nproc_per_node ${gpu_num} \
--
Gitblit v1.9.1