fix tester bug

phecda-xu · Jan 6, 2021 · 0237047 · 0237047
1 parent bef180c
commit 0237047
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tester/am_tester.py b/tester/am_tester.py
@@ -59,7 +59,7 @@ def _eval_step(self, batch):
                     i=i[:index]
                 index=j.index(self.text_featurizer.stop)
                 j=j[:index]
-            score, ws, wd, wi = wer(i, j)
+            score, ws, wd, wi = wer(j, i)
             self.cer_s+=ws
             self.cer_d+=wd
             self.cer_i+=wi

diff --git a/tester/lm_tester.py b/tester/lm_tester.py
@@ -38,7 +38,7 @@ def _eval_step(self, batch):
                 index = j.index(self.model.end_id)
                 j = j[1:index]
 
-            score, ws, wd, wi = wer(i, j)
+            score, ws, wd, wi = wer(j, i)
             self.cer_s+=ws
             self.cer_d+=wd
             self.cer_i+=wi