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 +++--------
funasr/models/transformer/model.py | 1 +
2 files changed, 4 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}" \
diff --git a/funasr/models/transformer/model.py b/funasr/models/transformer/model.py
index e813e22..a92f5af 100644
--- a/funasr/models/transformer/model.py
+++ b/funasr/models/transformer/model.py
@@ -118,6 +118,7 @@
# token_list, sym_space, sym_blank, report_cer, report_wer
# )
#
+ self.error_calculator = None
if ctc_weight == 0.0:
self.ctc = None
else:
--
Gitblit v1.9.1