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
ricardodeazambuja committed Apr 4, 2020
1 parent 607eb64 commit f1d3511
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 @@ -420,7 +420,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 f1d3511

Please sign in to comment.