[Logging] Save json file on wandb too

facebookresearch · Nov 27, 2023 · 792271f · 792271f
1 parent ee12a28
commit 792271f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/benchmarl/experiment/logger.py b/benchmarl/experiment/logger.py
@@ -214,6 +214,10 @@ def log_evaluation(
                 evaluation_step=total_frames
                 // self.experiment_config.evaluation_interval,
             )
+            for logger in self.loggers:
+                if isinstance(logger, WandbLogger):
+                    logger.experiment.save(str(self.json_writer.path))
+
         self.log(to_log, step=step)
         if video_frames is not None:
             vid = torch.tensor(