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: