Add the per epoch callback back

dcd76d49 · guptapriya · guptapriya · dbdf712e · dcd76d49
Commit dcd76d49 authored Jun 05, 2019 by guptapriya Committed by guptapriya Jun 04, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/official/recommendation/ncf_keras_main.py
+++ b/official/recommendation/ncf_keras_main.py
@@ -284,7 +284,8 @@ def run_ncf(_):
  eval_input_dataset = eval_input_dataset.batch(batches_per_step)

  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:
    early_stopping_callback = CustomEarlyStopping(