Skip to content

GitLab

Commit 719eec7b authored by guptapriya's avatar guptapriya
Browse files

Fix breakage due to early stopping callback

parent 6fc642d4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/recommendation/ncf_keras_main.py
View file @ 719eec7b
...@@ -156,6 +156,7 @@ class CustomEarlyStopping(tf.keras.callbacks.Callback): ...@@ -156,6 +156,7 @@ class CustomEarlyStopping(tf.keras.callbacks.Callback):
self.monitor = monitor self.monitor = monitor
self.desired = desired_value self.desired = desired_value
self.stopped_epoch = 0
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment