lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

In EarlyStopping, reset stopped_epoch in on_train_begin to allow it to be re-used (#6591)

Browse Source
This commit is contained in:
meberstein 2017-05-11 14:32:49 -07:00 committed by François Chollet
parent 24db6bfaaf
commit 5f4f234f9b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
keras/callbacks.py
View File

@ -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):