console logger fix

2020-05-19 16:44:37 +02:00 · 2020-05-19 16:44:37 +02:00 · 97cd39bf99
parent f75b0a6439
commit 97cd39bf99
1 changed files with 3 additions and 3 deletions
--- a/utils/console_logger.py
+++ b/utils/console_logger.py
@ -86,10 +86,10 @@ class ConsoleLogger():
            sign = '+'
            diff = 0
            if self.old_eval_loss_dict is not None:
-                diff = self.old_eval_loss_dict[key] - value
-                if diff > 0:
+                diff = value - self.old_eval_loss_dict[key]
+                if diff < 0:
                    color = tcolors.OKGREEN
-                    sign = '-'
+                    sign = ''
            log_text += "{}{}:{} {:.5f} {}({}{:.5f})\n".format(indent, key, color, value, tcolors.ENDC, sign, diff)
        self.old_eval_loss_dict = avg_loss_dict
        print(log_text, flush=True)