Skip to content

Commit

Permalink
Revert to pokemon exploration map
Browse files Browse the repository at this point in the history
thatguy11325 committed Mar 22, 2024
1 parent 5ebb02d commit cb7c675
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion pokemonred_puffer/cleanrl_puffer.py
Original file line number Diff line number Diff line change
@@ -464,7 +464,7 @@ def evaluate(self):
self.stats = {}
self.max_stats = {}
for k, v in self.infos["learner"].items():
if "exploration_map" in k and config.save_overlay is True:
if "pokemon_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:
2 changes: 1 addition & 1 deletion pokemonred_puffer/environment.py
Original file line number Diff line number Diff line change
@@ -635,7 +635,7 @@ def agent_stats(self, action):
},
"reward": self.get_game_state_reward(),
"reward/reward_sum": sum(self.get_game_state_reward().values()),
"exploration_map": self.explore_map,
"pokemon_exploration_map": self.explore_map,
"cut_exploration_map": self.cut_explore_map,
}

0 comments on commit cb7c675

Please sign in to comment.