Skip to content

GitLab

Commit d58f7590 authored by Zack Hsiao's avatar Zack Hsiao Committed by pkulzc
Browse files

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

parent 05383c7b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/model_main.py
View file @ d58f7590
...@@ -86,7 +86,7 @@ def main(unused_argv): ...@@ -86,7 +86,7 @@ def main(unused_argv):
input_fn = eval_input_fns[0] input_fn = eval_input_fns[0]
if FLAGS.run_once: if FLAGS.run_once:
estimator.evaluate(input_fn, estimator.evaluate(input_fn,
num_eval_steps=None, steps=None,
checkpoint_path=tf.train.latest_checkpoint( checkpoint_path=tf.train.latest_checkpoint(
FLAGS.checkpoint_dir)) FLAGS.checkpoint_dir))
else: else:
......
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