update metrics2str in tabular_logger.py

npurson · Apr 26, 2023 · 2e041e1 · 2e041e1
1 parent 37f830a
commit 2e041e1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tmpl/utils/tabular_logger.py b/tmpl/utils/tabular_logger.py
@@ -29,9 +29,9 @@ def log_metrics(self, metrics_dict: Dict[str, float], step: Optional[int] = None
 
         def metrics2str(metrics: dict) -> str:
             return ', '.join([
-                '{}: {{{}}}'.format(x, metrics2str(y)) if isinstance(y, dict) else
-                '{}: {:.5f}'.format(x, y) if isinstance(y, float) else '{}: {}'.format(x, y)
-                for x, y in metrics.items()
+                f'{x}: {{{metrics2str(y)}}}' if isinstance(y, dict) else
+                (f'{x}: {y:.4f}' if y >= 1e-4 else f'{x}: {y:.5f}')
+                if isinstance(y, float) else f'{x}: {y}' for x, y in metrics.items()
             ])
 
         log_str = metrics2str(metrics)