From 5fc8fdf9c1d02cf5fe056be4462120a5eceecfa6 Mon Sep 17 00:00:00 2001 From: Dmitry Lipin Date: Fri, 31 Jan 2020 18:49:44 +0300 Subject: [PATCH] fix import --- pytorch_lightning/loggers/tensorboard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/loggers/tensorboard.py b/pytorch_lightning/loggers/tensorboard.py index ee40f31f9b5a6..48459cd889e33 100644 --- 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)