update lock

awslabs · Jun 29, 2024 · 79fb0f5 · 79fb0f5
1 parent 066eb68
commit 79fb0f5
Show file tree

Hide file tree

Showing 3 changed files with 3,528 additions and 3,211 deletions.
diff --git a/fortuna/training/mixin.py b/fortuna/training/mixin.py
@@ -146,9 +146,7 @@ def early_stopping_update(
             early_stopping_monitor = validation_metrics[self.early_stopping_monitor]
             if self.early_stopping_mode == "max":
                 early_stopping_monitor = -early_stopping_monitor
-            self._early_stopping = self._early_stopping.update(
-                early_stopping_monitor
-            )
+            self._early_stopping = self._early_stopping.update(early_stopping_monitor)
             improved = self._early_stopping.has_improved
         return improved