Fix breakage due to early stopping callback

719eec7b · guptapriya · 6fc642d4 · 719eec7b
Commit 719eec7b authored May 28, 2019 by guptapriya
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

official/recommendation/ncf_keras_main.py official/recommendation/ncf_keras_main.py +1 -0

No files found.
--- a/official/recommendation/ncf_keras_main.py
+++ b/official/recommendation/ncf_keras_main.py
@@ -156,6 +156,7 @@ class CustomEarlyStopping(tf.keras.callbacks.Callback):

    self.monitor = monitor
    self.desired = desired_value
+    self.stopped_epoch = 0

  def on_epoch_end(self, epoch, logs=None):
    current = self.get_monitor_value(logs)