From c73d1a8e81582b91a9bdd6e82fce2e84f8d9d94b Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 14 八月 2023 19:31:55 +0800
Subject: [PATCH] update func cif_wo_hidden
---
egs/aishell2/transformerLM/run.sh | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/egs/aishell2/transformerLM/run.sh b/egs/aishell2/transformerLM/run.sh
index 28e3762..a5ea0ea 100755
--- a/egs/aishell2/transformerLM/run.sh
+++ b/egs/aishell2/transformerLM/run.sh
@@ -34,7 +34,7 @@
tag=exp1
model_dir="baseline_$(basename "${lm_config}" .yaml)_${lang}_${token_type}_${tag}"
lm_exp=${exp_dir}/exp/${model_dir}
-inference_lm=valid.loss.ave.pth # Language model path for decoding.
+inference_lm=valid.loss.ave.pb # Language model path for decoding.
stage=0
stop_stage=3
@@ -232,7 +232,7 @@
if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
echo "Stage 3: Calc perplexity: ${lm_test_text}"
- python ../../../funasr/bin/lm_inference.py \
+ python ../../../funasr/bin/lm_inference_launch.py \
--output_dir "${lm_exp}/perplexity_test" \
--ngpu "${gpu_num}" \
--batch_size 1 \
--
Gitblit v1.9.1