From 5f4f234f9b4b2edd78153a289cd6388a6330993d Mon Sep 17 00:00:00 2001 From: meberstein Date: Thu, 11 May 2017 14:32:49 -0700 Subject: [PATCH] In EarlyStopping, reset stopped_epoch in on_train_begin to allow it to be re-used (#6591) --- keras/callbacks.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/keras/callbacks.py b/keras/callbacks.py index 8cb1bc698..f111b1c08 100644 --- a/keras/callbacks.py +++ b/keras/callbacks.py @@ -482,7 +482,9 @@ class EarlyStopping(Callback): self.min_delta *= -1 def on_train_begin(self, logs=None): - self.wait = 0 # Allow instances to be re-used + # Allow instances to be re-used + self.wait = 0 + self.stopped_epoch = 0 self.best = np.Inf if self.monitor_op == np.less else -np.Inf def on_epoch_end(self, epoch, logs=None):