Merge pull request #38566 from eliberis:patch-1
PiperOrigin-RevId: 314263271 Change-Id: I7de744ff2405d22efdb1d61bf75b67c8b27c35bb
This commit is contained in:
commit
1a71abed71
@ -1627,6 +1627,7 @@ class EarlyStopping(Callback):
|
|||||||
self.best = self.baseline
|
self.best = self.baseline
|
||||||
else:
|
else:
|
||||||
self.best = np.Inf if self.monitor_op == np.less else -np.Inf
|
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):
|
def on_epoch_end(self, epoch, logs=None):
|
||||||
current = self.get_monitor_value(logs)
|
current = self.get_monitor_value(logs)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user