fix import

Lightning-AI · Feb 2, 2020 · 5fc8fdf · 5fc8fdf
1 parent fa50e1c
commit 5fc8fdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/loggers/tensorboard.py b/pytorch_lightning/loggers/tensorboard.py
@@ -6,7 +6,6 @@
 import torch
 import csv
 from torch.utils.tensorboard import SummaryWriter
-from torch.utils.tensorboard.summary import hparams
 
 from .base import LightningLoggerBase, rank_zero_only
 
@@ -85,6 +84,7 @@ def log_hyperparams(self, params):
                 " hyperparameter logging."
             )
         else:
+            from torch.utils.tensorboard.summary import hparams
             exp, ssi, sei = hparams(params, {})
             writer = self.experiment._get_file_writer()
             writer.add_summary(exp)