From b3fcd42bf6e60372ed0f0d3e53798fcdd713fe08 Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期三, 06 十二月 2023 11:14:57 +0800
Subject: [PATCH] Merge pull request #1144 from yeyupiaoling/main
---
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