upload the global map as jpg

thatguy11325 · Jun 27, 2024 · a2f2098 · a2f2098
1 parent 75dc986
commit a2f2098
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -309,7 +309,9 @@ def evaluate(self):
                             np.stack(self.infos["pokemon_exploration_map"], axis=0)
                         )
                         if self.wandb_client is not None:
-                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(overlay)
+                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(
+                                overlay, file_type="jpg"
+                            )
                     elif "state" in k:
                         continue