Merge pull request #33 from marcpinet/fix-callback-baseline-stop-trai…

…ning fix: none exception on iterable
marcpinet · Apr 24, 2024 · 153fb04 · 153fb04
2 parents 449d05f + 24a8cc7
commit 153fb04
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/neuralnetlib/model.py b/neuralnetlib/model.py
@@ -186,13 +186,13 @@ def fit(self, x_train: np.ndarray, y_train: np.ndarray, epochs: int, batch_size:
                         break
                     if callback.on_epoch_end(self, metrics_values):
                         break
+
+                if any(callback.stop_training for callback in callbacks):
+                                break
 
             if verbose:
                 print()
 
-            if any(callback.stop_training for callback in callbacks):
-                break
-
         if verbose:
             print()
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='neuralnetlib',
-    version='2.3.3',
+    version='2.3.4',
     author='Marc Pinet',
     description='A simple convolutional neural network library with only numpy as dependency',
     long_description=open('README.md', encoding="utf-8").read(),