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 Sep 16, 2018
1 parent 17fa528 commit 710aaff
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 @@ -387,7 +387,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(),
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)
Expand Down

0 comments on commit 710aaff

Please sign in to comment.