attempt to fix overlay?

thatguy11325 · Jun 26, 2024 · 580e69f · 580e69f
1 parent 6eed590
commit 580e69f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config.yaml b/config.yaml
@@ -22,7 +22,7 @@ debug:
     env_batch_size: 4
     env_pool: True
     zero_copy: False
-    batch_size: 128
+    batch_size: 4
     minibatch_size: 4
     batch_rows: 4
     bptt_horizon: 2
@@ -150,7 +150,6 @@ wrappers:
         jitter: 0
 
   stream_only:
-    - episode_stats.EpisodeStatsWrapper: {}
     - stream_wrapper.StreamWrapper:
         user: thatguy 
     - exploration.OnResetExplorationWrapper:
@@ -170,6 +169,7 @@ wrappers:
     - exploration.OnResetExplorationWrapper:
         full_reset_frequency: 25
         jitter: 0
+    - episode_stats.EpisodeStatsWrapper: {}
 
 rewards:
   baseline.BaselineRewardEnv:

diff --git a/pokemonred_puffer/eval.py b/pokemonred_puffer/eval.py
@@ -15,7 +15,7 @@ def make_pokemon_red_overlay(counts: np.ndarray):
     # The clip scaling needs to be re-calibrated since my
     # overlay is from the global map with fading
     scaled = np.ascontiguousarray(np.sum(counts, axis=0).astype(np.float32))
-    scaled = scaled / np.max(counts)
+    scaled = scaled / np.max(scaled)
     nonzero = np.ascontiguousarray(np.where(scaled > 0, 1, 0).astype(np.float32))
     # scaled = np.clip(counts, 0, 1000) / 1000.0