Skip to content

Commit b5cab7e

Browse files
authored
Create single file in TensorBoardLogger (#777)
* write to single file * fix import
1 parent 472f394 commit b5cab7e

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pytorch_lightning/loggers/tensorboard.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,12 @@ def log_hyperparams(self, params):
8484
" hyperparameter logging."
8585
)
8686
else:
87-
# `add_hparams` requires both - hparams and metric
88-
self.experiment.add_hparams(hparam_dict=params, metric_dict={})
87+
from torch.utils.tensorboard.summary import hparams
88+
exp, ssi, sei = hparams(params, {})
89+
writer = self.experiment._get_file_writer()
90+
writer.add_summary(exp)
91+
writer.add_summary(ssi)
92+
writer.add_summary(sei)
8993
# some alternative should be added
9094
self.tags.update(params)
9195

0 commit comments

Comments
 (0)