From eb3948b4805f58bf03d364e8a8c37c0ef3a854ca Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 16 五月 2023 11:11:44 +0800
Subject: [PATCH] update repo
---
egs/aishell/paraformerbert/run.sh | 17 +++++++++--------
1 files changed, 9 insertions(+), 8 deletions(-)
diff --git a/egs/aishell/paraformerbert/run.sh b/egs/aishell/paraformerbert/run.sh
index b46d4e4..a4ad195 100755
--- a/egs/aishell/paraformerbert/run.sh
+++ b/egs/aishell/paraformerbert/run.sh
@@ -111,12 +111,13 @@
world_size=$gpu_num # run on one machine
if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
echo "stage 3: Training"
-# if ! "${skip_extract_embed}"; then
-# echo "extract embeddings..."
-# local/extract_embeds.sh \
-# --bert_model_name ${bert_model_name} \
-# --raw_dataset_path ${feats_dir}
-# fi
+ if ! "${skip_extract_embed}"; then
+ echo "extract embeddings..."
+ local/extract_embeds.sh \
+ --bert_model_name ${bert_model_name} \
+ --raw_dataset_path ${feats_dir} \
+ --nj $nj
+ fi
mkdir -p ${exp_dir}/exp/${model_dir}
mkdir -p ${exp_dir}/exp/${model_dir}/log
INIT_FILE=${exp_dir}/exp/${model_dir}/ddp_init
@@ -139,6 +140,7 @@
--data_dir ${feats_dir}/data \
--train_set ${train_set} \
--valid_set ${valid_set} \
+ --embed_path ${feats_dir}/data \
--cmvn_file ${feats_dir}/data/${train_set}/cmvn/cmvn.mvn \
--speed_perturb ${speed_perturb} \
--resume true \
@@ -209,5 +211,4 @@
tail -n 3 ${_dir}/text.cer > ${_dir}/text.cer.txt
cat ${_dir}/text.cer.txt
done
-fi
-
+fi
\ No newline at end of file
--
Gitblit v1.9.1