From 0c64e61c97a8232203694c3672e254d736a13be3 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期三, 21 二月 2024 11:56:52 +0800
Subject: [PATCH] bug fix
---
examples/aishell/e_branchformer/run.sh | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/examples/aishell/e_branchformer/run.sh b/examples/aishell/e_branchformer/run.sh
index 0855798..ed765b8 100755
--- a/examples/aishell/e_branchformer/run.sh
+++ b/examples/aishell/e_branchformer/run.sh
@@ -109,6 +109,7 @@
log_file="${exp_dir}/exp/${model_dir}/train.log.txt.${current_time}"
echo "log_file: ${log_file}"
+ export CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
gpu_num=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
torchrun \
--nnodes 1 \
--
Gitblit v1.9.1