diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py index 9d15e69..893660f 100644 --- a/pokemonred_puffer/cleanrl_puffer.py +++ b/pokemonred_puffer/cleanrl_puffer.py @@ -319,9 +319,9 @@ def evaluate(self): # pull a state within that list new_state = random.choice(self.states[new_state_key]) """ - if self.config.train.early_stop: + if self.config.early_stop: to_delete = [] - for event, minutes in self.config.train.early_stop.values(): + for event, minutes in self.config.early_stop.values(): if any(event in key for key in self.state.keys()): to_delete.append(event) elif self.profile.uptime > minutes * 60 and all(