Fix WER transcription

e872b4ee · Yoach Lacombe · GitHub · 62d6bf69 · e872b4ee
Unverified Commit e872b4ee authored May 23, 2024 by Yoach Lacombe Committed by GitHub May 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

training/eval.py training/eval.py +1 -1

No files found.
--- a/training/eval.py
+++ b/training/eval.py
@@ -52,7 +52,7 @@ def wer(asr_model_name_or_path, prompts, audios, device, per_device_eval_batch_s
        if len(norm_ref) > 0:
            norm_pred = normalizer(pred["text"])
            normalized_predictions.append(norm_pred)
-            normalized_references.append(norm_pred)
+            normalized_references.append(norm_ref)

    word_error = 100 * metric.compute(predictions=normalized_predictions, references=normalized_references)