train: earlier check of stagnant loss

MPoL-dev · Nov 29, 2023 · c6e73e9 · c6e73e9
1 parent d8317db
commit c6e73e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mpol/training.py b/src/mpol/training.py
@@ -242,7 +242,7 @@ def train(self, model, dataset):
                 )
 
             # check early-on whether the loss isn't evolving
-            if count == 20:
+            if count == 10:# and self._scheduler is not None:
                 loss_arr = np.array(losses)
                 if all(0.9 <= loss_arr[:-1] / loss_arr[1:]) and all(
                     loss_arr[:-1] / loss_arr[1:] <= 1.1