Skip to content

Commit

Permalink
Update tensorboard.py
Browse files Browse the repository at this point in the history
  • Loading branch information
justusschock authored May 12, 2020
1 parent 1d4fb11 commit bc9ecd5
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions pytorch_lightning/loggers/tensorboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,9 +123,10 @@ def log_hyperparams(self, params: Union[Dict[str, Any], Namespace],
writer.add_summary(exp)
writer.add_summary(ssi)
writer.add_summary(sei)

# necessary for hparam comparison with metrics
self.log_metrics(metrics)

if metrics:
# necessary for hparam comparison with metrics
self.log_metrics(metrics)

# some alternative should be added
self.tags.update(sanitized_params)
Expand Down

0 comments on commit bc9ecd5

Please sign in to comment.