From 83a233e69197aac9ccf1292da5b074ec75be9b1a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 17:24:03 -0500 Subject: [PATCH] added loggers --- pytorch_lightning/logging/mlflow.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pytorch_lightning/logging/mlflow.py b/pytorch_lightning/logging/mlflow.py index 9a786db2..6dd77d5d 100644 --- a/pytorch_lightning/logging/mlflow.py +++ b/pytorch_lightning/logging/mlflow.py @@ -39,6 +39,15 @@ logger = getLogger(__name__) class MLFlowLogger(LightningLoggerBase): def __init__(self, experiment_name, tracking_uri=None, tags=None): + r""" + + Logs using MLFlow + + Args: + experiment_name (str): The name of the experiment + tracking_uri (str): where this should track + tags (dict): ? + """ super().__init__() self._mlflow_client = mlflow.tracking.MlflowClient(tracking_uri) self.experiment_name = experiment_name