bug fix

2020-06-12 11:37:32 +02:00 · 2020-06-12 11:37:32 +02:00 · 9af6bb0c29
parent 1c2dc9f739
commit 9af6bb0c29
1 changed files with 1 additions and 1 deletions
--- a/vocoder/utils/console_logger.py
+++ b/vocoder/utils/console_logger.py
@ -84,7 +84,7 @@ class ConsoleLogger():
            color = ''
            sign = '+'
            diff = 0
-            if self.old_eval_loss_dict is not None:
+            if self.old_eval_loss_dict is not None and key in self.old_eval_loss_dict:
                diff = value - self.old_eval_loss_dict[key]
                if diff < 0:
                    color = tcolors.OKGREEN