config fixing

thatguy11325 · Oct 26, 2024 · 4446570 · 4446570
1 parent 595b4bd
commit 4446570
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(