wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Hacky fix for mlflow logger (#277)

Browse Source 
* Hacky fix for mlflow logger

It dies when "created_at" is logged

* Log warning
This commit is contained in:
Anton Bakhtin
2019-10-01 21:32:52 -04:00
committed by William Falcon
parent 133d6b3ec1
commit 222d7d2d5d
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/logging/mlflow_logger.py
+5
View File
@@ -34,6 +34,11 @@ class MLFlowLogger(LightningLoggerBase):
def log_metrics(self, metrics, step_num=None): def log_metrics(self, metrics, step_num=None):
timestamp_ms = int(time() * 1000) timestamp_ms = int(time() * 1000)
for k, v in metrics.items(): 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) self.client.log_metric(self.run_id, k, v, timestamp_ms, step_num)
def save(self): def save(self):