| | |
|---|
| | | gpuid_list="0,1" # set gpus, e.g., gpuid_list="0,1" |
|---|
| | | njob=4 # the number of jobs for CPU decoding, if gpu_inference=false, use CPU decoding, please set njob |
|---|
| | | |
|---|
| | | . utils/parse_options.sh || exit 1; |
|---|
| | | |
|---|
| | | if ${gpu_inference}; then |
|---|
| | | if ${gpu_inference} == "true"; then |
|---|
| | | nj=$(echo $gpuid_list | awk -F "," '{print NF}') |
|---|
| | | else |
|---|
| | | nj=$njob |
|---|
| | |
|---|
| | | |
|---|
| | | if [ $stage -le 2 ] && [ $stop_stage -ge 2 ];then |
|---|
| | | echo "Computing WER ..." |
|---|
| | | python utils/proce_text.py ${output_dir}/1best_recog/text ${output_dir}/1best_recog/text.proc |
|---|
| | | python utils/proce_text.py ${data_dir}/text ${output_dir}/1best_recog/text.ref |
|---|
| | | cp ${output_dir}/1best_recog/text ${output_dir}/1best_recog/text.proc |
|---|
| | | cp ${data_dir}/text ${output_dir}/1best_recog/text.ref |
|---|
| | | python utils/compute_wer.py ${output_dir}/1best_recog/text.ref ${output_dir}/1best_recog/text.proc ${output_dir}/1best_recog/text.cer |
|---|
| | | tail -n 3 ${output_dir}/1best_recog/text.cer |
|---|
| | | fi |
|---|
