wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix hparam logging with metrics (#1647)

Browse Source 
* add metric logging

* Use pytorch built-in method

* Update tensorboard.py

* Update tensorboard.py
This commit is contained in:
Justus Schock
2020-05-12 07:25:12 -04:00
committed by GitHub
parent 35ac30e688
commit 5f292390fd
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/loggers/tensorboard.py
+5
View File
@@ -115,6 +115,7 @@ class TensorBoardLogger(LightningLoggerBase):
)
else:
from torch.utils.tensorboard.summary import hparams
if metrics is None:
metrics = {}
exp, ssi, sei = hparams(sanitized_params, metrics)
@@ -123,6 +124,10 @@ class TensorBoardLogger(LightningLoggerBase):
writer.add_summary(ssi)
writer.add_summary(sei)
if metrics:
# necessary for hparam comparison with metrics
self.log_metrics(metrics)
# some alternative should be added
self.tags.update(sanitized_params)