From 7dc937f063d6b67bca070a8d7c0368ec01bfb96c Mon Sep 17 00:00:00 2001 From: Edgar Liberis Date: Wed, 15 Apr 2020 09:39:32 +0100 Subject: [PATCH] Reset best weights in EarlyStopping.on_train_begin EarlyStopping callback instances can be reused in different training loops, but the `on_train_begin` hook forgets to reset the `self.best_weights` state. --- tensorflow/python/keras/callbacks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/python/keras/callbacks.py b/tensorflow/python/keras/callbacks.py index 2d178bb6409..e0cea492664 100644 --- a/tensorflow/python/keras/callbacks.py +++ b/tensorflow/python/keras/callbacks.py @@ -1524,6 +1524,7 @@ class EarlyStopping(Callback): self.best = self.baseline else: self.best = np.Inf if self.monitor_op == np.less else -np.Inf + self.best_weights = None def on_epoch_end(self, epoch, logs=None): current = self.get_monitor_value(logs)