Update trainer

yebiny · Feb 3, 2021 · f53ea49 · f53ea49
1 parent c258889
commit f53ea49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kospeech/trainer/supervised_trainer.py b/kospeech/trainer/supervised_trainer.py
@@ -262,7 +262,7 @@ def _train_epoches(
                 architecture=architecture,
             )
 
-            if architecture in ('rnnt', 'conformer_t'):
+            if architecture not in ('rnnt', 'conformer_t'):
                 y_hats = output.max(-1)[1]
                 cer = self.metric(targets, y_hats)