Merge pull request #38566 from eliberis:patch-1

PiperOrigin-RevId: 314263271
Change-Id: I7de744ff2405d22efdb1d61bf75b67c8b27c35bb
This commit is contained in:
TensorFlower Gardener 2020-06-01 21:34:02 -07:00
commit 1a71abed71

View File

@ -1627,6 +1627,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)