Skip to content

Commit 7ad5c45

Browse files
add WER for RNN-T (#440)
1 parent f0bd9f9 commit 7ad5c45

File tree

1 file changed

+2
-0
lines changed
  • quickstart/language_modeling/pytorch/rnnt/inference/cpu

1 file changed

+2
-0
lines changed

quickstart/language_modeling/pytorch/rnnt/inference/cpu/accuracy.sh

+2
Original file line numberDiff line numberDiff line change
@@ -75,4 +75,6 @@ python -m intel_extension_for_pytorch.cpu.launch \
7575
wait
7676

7777
accuracy=$(grep 'Accuracy:' ${OUTPUT_DIR}/rnnt_${PRECISION}_inference_accuracy* |sed -e 's/.*Accuracy//;s/[^0-9.]//g')
78+
WER=$(grep 'Evaluation WER:' ${OUTPUT_DIR}/rnnt_${PRECISION}_inference_accuracy* |sed -e 's/.*Evaluation WER//;s/[^0-9.]//g')
7879
echo ""RNN-T";"accuracy";$1; ${BATCH_SIZE};${accuracy}" | tee -a ${OUTPUT_DIR}/summary.log
80+
echo ""RNN-T";"WER";$1; ${BATCH_SIZE};${WER}" | tee -a ${work_space}/summary.log

0 commit comments

Comments
 (0)