From 7963802e6ae4f86d85776b352da69b8fcee9481c Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Fri, 21 Jun 2024 14:22:50 -0400 Subject: [PATCH] fix log loop --- pokemonred_puffer/cleanrl_puffer.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py index e6ac5cc..91fe6ad 100644 --- a/pokemonred_puffer/cleanrl_puffer.py +++ b/pokemonred_puffer/cleanrl_puffer.py @@ -297,13 +297,8 @@ def evaluate(self): ) if self.wandb_client is not None: self.stats["Media/aggregate_exploration_map"] = wandb.Image(overlay) - - for k, v in self.infos.items(): - if "_map" in k and self.wandb_client is not None: - self.stats[f"Media/{k}"] = wandb.Image(v[0]) - continue elif "state" in k: - pass + continue try: # TODO: Better checks on log data types self.stats[k] = np.mean(v)