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/e_branchformer/run.sh | 13 ++++++++-----
1 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/examples/aishell/e_branchformer/run.sh b/examples/aishell/e_branchformer/run.sh
index ce6d346..452ec80 100755
--- a/examples/aishell/e_branchformer/run.sh
+++ b/examples/aishell/e_branchformer/run.sh
@@ -1,11 +1,11 @@
#!/usr/bin/env bash
-CUDA_VISIBLE_DEVICES="0,1"
+CUDA_VISIBLE_DEVICES="0,1,2,3"
# general configuration
feats_dir="../DATA" #feature output dictionary
-exp_dir="."
+exp_dir=`pwd`
lang=zh
token_type=char
stage=0
@@ -15,9 +15,9 @@
nj=32
inference_device="cuda" #"cpu"
-inference_checkpoint="model.pt"
+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;
@@ -78,7 +80,7 @@
--config-name "${config}" \
++train_data_set_list="${feats_dir}/data/${train_set}/audio_datasets.jsonl" \
++cmvn_file="${feats_dir}/data/${train_set}/cmvn.json" \
- ++dataset_conf.num_workers=$nj
+
fi
token_list=${feats_dir}/data/${lang}_token_list/$token_type/tokens.txt
@@ -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