diff --git a/utils/console_logger.py b/utils/console_logger.py index 5c6ec75f..0b361bb8 100644 --- a/utils/console_logger.py +++ b/utils/console_logger.py @@ -87,9 +87,9 @@ class ConsoleLogger(): diff = 0 if self.old_eval_loss_dict is not None: diff = value - self.old_eval_loss_dict[key] - if diff < 0: + if diff <= 0: color = tcolors.OKGREEN 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 + print(log_text, flush=True)