mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
added loggers
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user