Skip to content

GitLab

Commit dcd76d49 authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

Add the per epoch callback back

parent dbdf712e
Hide 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