Added missing metrics when logging on tensorboard (#1298)

Co-authored-by: Riccardo Sepe <[email protected]> Co-authored-by: Francesco Scalera <[email protected]>
DLR-RM · Jan 26, 2023 · bed8f48 · bed8f48
1 parent 5ee9009
commit bed8f48
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/stable_baselines3/common/logger.py b/stable_baselines3/common/logger.py
@@ -430,6 +430,8 @@ def write(self, key_values: Dict[str, Any], key_excluded: Dict[str, Union[str, T
                 self.writer.file_writer.add_summary(experiment)
                 self.writer.file_writer.add_summary(session_start_info)
                 self.writer.file_writer.add_summary(session_end_info)
+                for k, v in value.metric_dict.items():
+                    self.writer.file_writer.add_scalar(k, v)
 
         # Flush the output to the file
         self.writer.flush()