Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit dcd76d49 authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

Add the per epoch callback back

parent dbdf712e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/recommendation/ncf_keras_main.py
View file @ dcd76d49
...@@ -284,7 +284,8 @@ def run_ncf(_): ...@@ -284,7 +284,8 @@ def run_ncf(_):
eval_input_dataset = eval_input_dataset.batch(batches_per_step) eval_input_dataset = eval_input_dataset.batch(batches_per_step)
time_callback = keras_utils.TimeHistory(batch_size, FLAGS.log_steps) time_callback = keras_utils.TimeHistory(batch_size, FLAGS.log_steps)
callbacks = [time_callback] per_epoch_callback = IncrementEpochCallback(producer)
callbacks = [per_epoch_callback, time_callback]
if FLAGS.early_stopping: if FLAGS.early_stopping:
early_stopping_callback = CustomEarlyStopping( early_stopping_callback = CustomEarlyStopping(
......
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