Fix Object detection API wrong argument when using run_once (#6010)

d58f7590 · Zack Hsiao · pkulzc · 05383c7b · d58f7590
Commit d58f7590 authored Feb 10, 2019 by Zack Hsiao Committed by pkulzc Feb 09, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/model_main.py research/object_detection/model_main.py +1 -1

No files found.
--- a/research/object_detection/model_main.py
+++ b/research/object_detection/model_main.py
@@ -86,7 +86,7 @@ def main(unused_argv):
      input_fn = eval_input_fns[0]
    if FLAGS.run_once:
      estimator.evaluate(input_fn,
-                         num_eval_steps=None,
+                         steps=None,
                         checkpoint_path=tf.train.latest_checkpoint(
                             FLAGS.checkpoint_dir))
    else: