wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-04 13:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

added loggers

Browse Source
This commit is contained in:
William Falcon committed 2020-01-21 14:07:44 -05:00
1 parent 94943afcb8
commit aae963a186
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/logging/mlflow.py
+1
View File
@@ -47,6 +47,7 @@ class MLFlowLogger(LightningLoggerBase):
experiment_name (str): The name of the experiment
tracking_uri (str): where this should track
tags (dict): todo this param
"""
super().__init__()
self._mlflow_client = mlflow.tracking.MlflowClient(tracking_uri)