fix import

Lightning-AI · Jan 31, 2020 · 64620a0 · 64620a0
1 parent a5288d0
commit 64620a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/logging/tensorboard.py b/pytorch_lightning/logging/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)