From 2a66366be4c2715870e4859fd5a5db6e8a9dc00a Mon Sep 17 00:00:00 2001
From: chenmengzheAAA <123789350+chenmengzheAAA@users.noreply.github.com>
Date: 星期四, 14 九月 2023 19:00:17 +0800
Subject: [PATCH] Merge pull request #956 from alibaba-damo-academy/chenmengzheAAA-patch-4
---
egs/aishell2/transformerLM/run.sh | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/egs/aishell2/transformerLM/run.sh b/egs/aishell2/transformerLM/run.sh
index 4a647c9..b04c643 100755
--- a/egs/aishell2/transformerLM/run.sh
+++ b/egs/aishell2/transformerLM/run.sh
@@ -214,13 +214,15 @@
echo "Stage 3: Calc perplexity: ${lm_test_text}"
python ../../../funasr/bin/lm_inference_launch.py \
- --output_dir "${lm_exp}/perplexity_test" \
+ --output_dir "${lm_exp}/perplexity_test/output.1" \
--ngpu "${gpu_num}" \
--batch_size 1 \
--train_config "${lm_exp}"/config.yaml \
--model_file "${lm_exp}/${inference_lm}" \
--data_path_and_name_and_type "${lm_test_text},text,text" \
--num_workers 1 \
+ --gpuid_list 0 \
+ --mode "transformer" \
--split_with_space false
fi
--
Gitblit v1.9.1