From b62ad06f4cf67cb1db5da20c72ea1bda7819f33a Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Fri, 12 Jun 2020 11:37:32 +0200
Subject: [PATCH] bug fix

---
 vocoder/utils/console_logger.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vocoder/utils/console_logger.py b/vocoder/utils/console_logger.py
index 50882160..4f44986c 100644
--- 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