passing experiment to wandb (#720)

Author: colehurwitz

pytorch_lightning/logging/wandb.py

@@ -33,15 +33,15 @@ class WandbLogger(LightningLoggerBase):
     """
 
     def __init__(self, name=None, save_dir=None, offline=False, id=None, anonymous=False,
-                 version=None, project=None, tags=None):
+                 version=None, project=None, tags=None, experiment=None):
         super().__init__()
         self._name = name
         self._save_dir = save_dir
         self._anonymous = "allow" if anonymous else None
         self._id = version or id
         self._tags = tags
         self._project = project
-        self._experiment = None
+        self._experiment = experiment
         self._offline = offline
 
     def __getstate__(self):
@@ -81,7 +81,7 @@ def log_hyperparams(self, params):
     @rank_zero_only
     def log_metrics(self, metrics, step=None):
         metrics["global_step"] = step
-        self.experiment.history.add(metrics)
+        self.experiment.log(metrics)
 
     def save(self):
         pass