Skip to content

Commit e2f1043

Browse files
rogierzRiccardo Sepefrancescoscalera99
committed
Added missing metrics when logging on tensorboard (#1298)
Co-authored-by: Riccardo Sepe <pastarick@users.noreply.github.com> Co-authored-by: Francesco Scalera <francescoscalera99@users.noreply.github.com>
1 parent 12e9917 commit e2f1043

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

stable_baselines3/common/logger.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,8 @@ def write(self, key_values: Dict[str, Any], key_excluded: Dict[str, Union[str, T
427427
self.writer.file_writer.add_summary(experiment)
428428
self.writer.file_writer.add_summary(session_start_info)
429429
self.writer.file_writer.add_summary(session_end_info)
430+
for k, v in value.metric_dict.items():
431+
self.writer.add_scalar(k, v)
430432

431433
# Flush the output to the file
432434
self.writer.flush()

0 commit comments

Comments
 (0)