From d44cf7b1cab683a8aa3876619c82226f4e6d6f3b Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Mon, 18 Sep 2023 18:11:33 +0200
Subject: fix

---
 swr2_asr/train.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'swr2_asr/train.py')

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index 1e57ba0..5277c16 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -142,8 +142,10 @@ def test(test_args: TestArgs) -> tuple[float, float, float]:
             for j, _ in enumerate(decoded_preds):
                 if j >= len(decoded_targets):
                     break
-                test_cer.append(cer(decoded_targets[j], decoded_preds[j][0].words[0]))
-                test_wer.append(wer(decoded_targets[j], decoded_preds[j][0].words[0]))
+                pred = " ".join(decoded_preds[j][0].words).strip()  # batch, top, words
+                target = decoded_targets[j]
+                test_cer.append(cer(target, pred))
+                test_wer.append(wer(target, pred))
 
     avg_cer = sum(test_cer) / len(test_cer)
     avg_wer = sum(test_wer) / len(test_wer)
-- 
cgit v1.2.3