Comment out cut related image logging

thatguy11325 · Jun 2, 2024 · aabe87a · aabe87a
1 parent 9cc652d
commit aabe87a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git 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
@@ -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: