From 222d7d2d5d01771e63449bdcfcdaf7dd8ed1ecec Mon Sep 17 00:00:00 2001 From: Anton Bakhtin <7470747+akhti@users.noreply.github.com> Date: Tue, 1 Oct 2019 21:32:52 -0400 Subject: [PATCH] Hacky fix for mlflow logger (#277) * Hacky fix for mlflow logger It dies when "created_at" is logged * Log warning --- pytorch_lightning/logging/mlflow_logger.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pytorch_lightning/logging/mlflow_logger.py b/pytorch_lightning/logging/mlflow_logger.py index c9cf3940..cc76ec4d 100644 --- a/pytorch_lightning/logging/mlflow_logger.py +++ b/pytorch_lightning/logging/mlflow_logger.py @@ -34,6 +34,11 @@ class MLFlowLogger(LightningLoggerBase): def log_metrics(self, metrics, step_num=None): timestamp_ms = int(time() * 1000) for k, v in metrics.items(): + if isinstance(v, str): + logger.warning( + f"Discarding metric with string value {k}={v}" + ) + continue self.client.log_metric(self.run_id, k, v, timestamp_ms, step_num) def save(self):