From aabe87ac673504caf51643d48cb3d705fe090148 Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Sat, 1 Jun 2024 21:11:30 -0400
Subject: [PATCH] Comment out cut related image logging

---
 config.yaml                         | 10 +++++-----
 pokemonred_puffer/cleanrl_puffer.py | 14 +++++++-------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/config.yaml b/config.yaml
index 530ecee..f30be7b 100644
--- a/config.yaml
+++ b/config.yaml
@@ -28,8 +28,8 @@ debug:
     env_pool: False
     log_frequency: 5000
     load_optimizer_state: False
-    swarm_frequency: 10
-    swarm_keep_pct: .1
+    # swarm_frequency: 10
+    # swarm_keep_pct: .1
 
 env:
   headless: True
@@ -84,13 +84,13 @@ train:
   save_checkpoint: False
   checkpoint_interval: 200
   save_overlay: True
-  overlay_interval: 200
+  overlay_interval: 100
   cpu_offload: True
   pool_kernel: [0]
   load_optimizer_state: False
 
-  swarm_frequency: 500
-  swarm_keep_pct: .8
+  # swarm_frequency: 500
+  # swarm_keep_pct: .8
 
 wrappers:
   baseline:
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
index b74b193..3c62d3e 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -524,13 +524,13 @@ def evaluate(self):
                     overlay = make_pokemon_red_overlay(np.stack(v, axis=0))
                     if self.wandb is not None:
                         self.stats["Media/aggregate_exploration_map"] = self.wandb.Image(overlay)
-            elif "cut_exploration_map" in k and config.save_overlay is True:
-                if self.update % config.overlay_interval == 0:
-                    overlay = make_pokemon_red_overlay(np.stack(v, axis=0))
-                    if self.wandb is not None:
-                        self.stats["Media/aggregate_cut_exploration_map"] = self.wandb.Image(
-                            overlay
-                        )
+            # elif "cut_exploration_map" in k and config.save_overlay is True:
+            #     if self.update % config.overlay_interval == 0:
+            #         overlay = make_pokemon_red_overlay(np.stack(v, axis=0))
+            #         if self.wandb is not None:
+            #             self.stats["Media/aggregate_cut_exploration_map"] = self.wandb.Image(
+            #                 overlay
+            #             )
             elif "state" in k:
                 pass
             else: