From c568628130ac42ebeea8cf48fe926520a31ff511 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 16 五月 2023 10:57:21 +0800
Subject: [PATCH] update repo

---
 egs/aishell/paraformerbert/run.sh |   16 ++++++++--------
 1 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/egs/aishell/paraformerbert/run.sh b/egs/aishell/paraformerbert/run.sh
index b46d4e4..e65ad2c 100755
--- a/egs/aishell/paraformerbert/run.sh
+++ b/egs/aishell/paraformerbert/run.sh
@@ -111,12 +111,12 @@
 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}
+    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 +139,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 +210,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