From 0a4fa37891f6d909f9f6463537b1261daec0e7f1 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 10 六月 2024 00:40:25 +0800
Subject: [PATCH] fix bug

---
 examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh |   24 ++++++++++++------------
 1 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
index 81f211f..c4e7afe 100644
--- a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
+++ b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
@@ -13,8 +13,8 @@
     jsonl=${jsonl_dir}/${data_set}
     output_dir=${out_dir}/${data_set}
     mkdir -p ${output_dir}
-    pred_file=${output_dir}/${data_set}/1best_recog/text_tn
-    ref_file=${output_dir}/${data_set}/1best_recog/label
+    pred_file=${output_dir}/1best_recog/text_tn
+    ref_file=${output_dir}/1best_recog/label
 
     python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir} ${device}
 
@@ -27,10 +27,10 @@
     jsonl=${jsonl_dir}/${data_set}
     output_dir=${out_dir}/${data_set}
     mkdir -p ${output_dir}
-    pred_file=${output_dir}/${data_set}/1best_recog/text_tn
-    ref_file=${output_dir}/${data_set}/1best_recog/label
+    pred_file=${output_dir}/1best_recog/text_tn
+    ref_file=${output_dir}/1best_recog/label
 
-    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir}
+    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir} ${device}
 
     python /mnt/workspace/zhifu.gzf/codebase/FunASR/funasr/metrics/wer.py ++ref_file=${ref_file} ++hyp_file=${pred_file} ++cer_file=${pred_file}.cer ++cn_postprocess=true
 
@@ -40,10 +40,10 @@
     jsonl=${jsonl_dir}/${data_set}
     output_dir=${out_dir}/${data_set}
     mkdir -p ${output_dir}
-    pred_file=${output_dir}/${data_set}/1best_recog/text_tn
-    ref_file=${output_dir}/${data_set}/1best_recog/label
+    pred_file=${output_dir}/1best_recog/text_tn
+    ref_file=${output_dir}/1best_recog/label
 
-    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir}
+    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir} ${device}
 
     python /mnt/workspace/zhifu.gzf/codebase/FunASR/funasr/metrics/wer.py ++ref_file=${ref_file} ++hyp_file=${pred_file} ++cer_file=${pred_file}.cer ++cn_postprocess=true
 
@@ -53,11 +53,11 @@
     jsonl=${jsonl_dir}/${data_set}
     output_dir=${out_dir}/${data_set}
     mkdir -p ${output_dir}
-    pred_file=${output_dir}/${data_set}/1best_recog/text_tn
-    ref_file=${output_dir}/${data_set}/1best_recog/label
+    pred_file=${output_dir}/1best_recog/text_tn
+    ref_file=${output_dir}/1best_recog/label
 
-    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir}
+    python ./demo_speech2text.py ${ckpt_dir} ${ckpt_id} ${jsonl} ${output_dir} ${device}
 
-    python /mnt/workspace/zhifu.gzf/codebase/FunASR/funasr/metrics/wer.py ++ref_file=${ref_file} ++hyp_file=${pred_file} ++cer_file=${pred_file}.cer ++cn_postprocess=true
+    python /mnt/workspace/zhifu.gzf/codebase/FunASR/funasr/metrics/wer.py ++ref_file=${ref_file} ++hyp_file=${pred_file} ++cer_file=${pred_file}.cer ++cn_postprocess=false
 
 done
\ No newline at end of file

--
Gitblit v1.9.1