From 97cd39bf99cd954a865200e4777b117dbff0ac8a Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 19 May 2020 16:44:37 +0200 Subject: [PATCH] console logger fix --- utils/console_logger.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/console_logger.py b/utils/console_logger.py index 5a37ac10..5c6ec75f 100644 --- 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) \ No newline at end of file