This commit is contained in:
lyblsgo 2023-09-11 16:12:52 +08:00
commit 24fd78d315
2 changed files with 4 additions and 1 deletions

View File

@ -214,13 +214,15 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
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

View File

@ -183,6 +183,7 @@ def inference_lm(
dtype=dtype,
batch_size=batch_size,
key_file=key_file,
preprocess_fn=preprocessor,
num_workers=num_workers,
)