Minor change, removed exit() after early stopping.

MarcBS · Sep 10, 2019 · 9c5a28e · 9c5a28e
1 parent 1f253bd
commit 9c5a28e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_wrapper/extra/callbacks.py b/keras_wrapper/extra/callbacks.py
@@ -842,7 +842,7 @@ def evaluate(self, epoch, counter_name='epoch'):
                         str(counter_name), epoch, self.metric_check, str(counter_name), self.best_epoch,
                         self.best_score if not self.want_to_minimize else -self.best_score))
                 self.model.stop_training = True
-                exit(1)
+                #exit(1)
 
 
 class LearningRateReducer(KerasCallback):