Fix crash when --eval_batch_size is not set. (#4955)

e4034bef · Reed · Taylor Robie · a2a943da · e4034bef
Commit e4034bef authored Jul 31, 2018 by Reed Committed by Taylor Robie Jul 31, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/recommendation/ncf_main.py official/recommendation/ncf_main.py +1 -1

No files found.
--- a/official/recommendation/ncf_main.py
+++ b/official/recommendation/ncf_main.py
@@ -202,7 +202,7 @@ def run_ncf(_):
  num_gpus = flags_core.get_num_gpus(FLAGS)
  batch_size = distribution_utils.per_device_batch_size(
      int(FLAGS.batch_size), num_gpus)
-  eval_batch_size = int(FLAGS.eval_batch_size) or int(FLAGS.batch_size)
+  eval_batch_size = int(FLAGS.eval_batch_size or FLAGS.batch_size)
  ncf_dataset = data_preprocessing.instantiate_pipeline(
      dataset=FLAGS.dataset, data_dir=FLAGS.data_dir,
      batch_size=batch_size,