From 114282885589956a29d7bcd015f55e966cb12532 Mon Sep 17 00:00:00 2001
From: Asaf Agami <asaf92@gmail.com>
Date: Sun, 10 Sep 2023 18:09:38 +0300
Subject: [PATCH] fix: model does not stop on stop_training == true

---
 src/TensorFlowNET.Keras/Engine/Model.Fit.cs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/TensorFlowNET.Keras/Engine/Model.Fit.cs b/src/TensorFlowNET.Keras/Engine/Model.Fit.cs
index de57f19ae..d6f89d8be 100644
--- a/src/TensorFlowNET.Keras/Engine/Model.Fit.cs
+++ b/src/TensorFlowNET.Keras/Engine/Model.Fit.cs
@@ -224,6 +224,10 @@ History FitInternal(DataHandler data_handler, int epochs, int validation_step, i
 
                 GC.Collect();
                 GC.WaitForPendingFinalizers();
+                if (stop_training)
+                {
+                    break;
+                }
             }
 
             return callbacks.History;
@@ -283,6 +287,10 @@ History FitInternal(DataHandler data_handler, int epochs, int verbose, List<ICal
 
                 GC.Collect();
                 GC.WaitForPendingFinalizers();
+                if (stop_training)
+                {
+                    break;
+                }
             }
 
             return callbacks.History;
@@ -339,6 +347,10 @@ History FitInternal(DataHandler data_handler, int epochs, int verbose, List<ICal
 
                 GC.Collect();
                 GC.WaitForPendingFinalizers();
+                if (stop_training)
+                {
+                    break;
+                }
             }
 
             return callbacks.History;