Unverified Commit 43f88f91 authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #4521 from aeloyq/master

Fixing bug for evaluating the quantized model in research/object_detection
parents cac3a298 b4d3eaab
...@@ -239,12 +239,12 @@ def evaluate(create_input_dict_fn, create_model_fn, eval_config, categories, ...@@ -239,12 +239,12 @@ def evaluate(create_input_dict_fn, create_model_fn, eval_config, categories,
keep_image_id_for_visualization_export) keep_image_id_for_visualization_export)
return result_dict, result_losses_dict return result_dict, result_losses_dict
if graph_hook_fn: graph_hook_fn()
variables_to_restore = tf.global_variables() variables_to_restore = tf.global_variables()
global_step = tf.train.get_or_create_global_step() global_step = tf.train.get_or_create_global_step()
variables_to_restore.append(global_step) variables_to_restore.append(global_step)
if graph_hook_fn: graph_hook_fn()
if eval_config.use_moving_averages: if eval_config.use_moving_averages:
variable_averages = tf.train.ExponentialMovingAverage(0.0) variable_averages = tf.train.ExponentialMovingAverage(0.0)
variables_to_restore = variable_averages.variables_to_restore() variables_to_restore = variable_averages.variables_to_restore()
......
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