added loggers

This commit is contained in:
William Falcon
2020-01-16 17:24:41 -05:00
parent 83a233e691
commit 48d1df042d
+1 -1
View File
@@ -46,7 +46,7 @@ class MLFlowLogger(LightningLoggerBase):
Args:
experiment_name (str): The name of the experiment
tracking_uri (str): where this should track
tags (dict): ?
tags (dict): todo this param
"""
super().__init__()
self._mlflow_client = mlflow.tracking.MlflowClient(tracking_uri)