lyblsgo
2023-09-11 24fd78d315f1e3928b919906cbb690e2995ef2c0
Merge branch 'main' of https://github.com/alibaba-damo-academy/FunASR
2个文件已修改
5 ■■■■ 已修改文件
egs/aishell2/transformerLM/run.sh 4 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/lm_inference_launch.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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
funasr/bin/lm_inference_launch.py
@@ -183,6 +183,7 @@
            dtype=dtype,
            batch_size=batch_size,
            key_file=key_file,
            preprocess_fn=preprocessor,
            num_workers=num_workers,
        )