From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var),由于mean和var是numpy,导致写入到文件的格式错误,参考上面的话,大概率是list(mean.tolist()),其实外层list没有必要 (#2437)
---
examples/aishell/transformer/run.sh | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/examples/aishell/transformer/run.sh b/examples/aishell/transformer/run.sh
index 336c558..3fb8465 100755
--- a/examples/aishell/transformer/run.sh
+++ b/examples/aishell/transformer/run.sh
@@ -15,9 +15,9 @@
nj=32
inference_device="cuda" #"cpu"
-inference_checkpoint="model.pt.avg5"
+inference_checkpoint="model.pt.avg10"
inference_scp="wav.scp"
-inference_batch_size=32
+inference_batch_size=1
# data
raw_data=../raw_data
@@ -26,6 +26,8 @@
# exp tag
tag="exp1"
workspace=`pwd`
+
+master_port=12345
. utils/parse_options.sh || exit 1;
@@ -115,6 +117,7 @@
torchrun \
--nnodes 1 \
--nproc_per_node ${gpu_num} \
+ --master_port ${master_port} \
../../../funasr/bin/train.py \
--config-path "${workspace}/conf" \
--config-name "${config}" \
--
Gitblit v1.9.1