Skip to content

Commit

Permalink
https://github.com/tensorflow/models/issues/4780#issuecomment-405441448
Browse files Browse the repository at this point in the history
  • Loading branch information
david-macleod committed Nov 2, 2018
1 parent 31ae57e commit dee188b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion research/object_detection/model_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -415,7 +415,7 @@ def tpu_scaffold():

# Eval metrics on a single example.
eval_metric_ops = eval_util.get_eval_metric_ops_for_evaluators(
eval_config, category_index.values(), eval_dict)
eval_config, list(category_index.values()), eval_dict)
for loss_key, loss_tensor in iter(losses_dict.items()):
eval_metric_ops[loss_key] = tf.metrics.mean(loss_tensor)
for var in optimizer_summary_vars:
Expand Down

0 comments on commit dee188b

Please sign in to comment.