From a2f2098d6055267e0009dcf39ef0a03f4cf0acbe Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Thu, 27 Jun 2024 13:52:20 -0400 Subject: [PATCH] upload the global map as jpg --- pokemonred_puffer/cleanrl_puffer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py index aa6e3fe..c8230a3 100644 --- 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