Create single file in TensorBoardLogger (#777)

* write to single file * fix import
Lightning-AI · Feb 2, 2020 · b5cab7e · b5cab7e
1 parent 472f394
commit b5cab7e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pytorch_lightning/loggers/tensorboard.py b/pytorch_lightning/loggers/tensorboard.py
@@ -84,8 +84,12 @@ def log_hyperparams(self, params):
                 " hyperparameter logging."
             )
         else:
-            # `add_hparams` requires both - hparams and metric
-            self.experiment.add_hparams(hparam_dict=params, metric_dict={})
+            from torch.utils.tensorboard.summary import hparams
+            exp, ssi, sei = hparams(params, {})
+            writer = self.experiment._get_file_writer()
+            writer.add_summary(exp)
+            writer.add_summary(ssi)
+            writer.add_summary(sei)
         # some alternative should be added
         self.tags.update(params)