From 33d340a5c6e53922241ed4ab31ac0ee270740ed7 Mon Sep 17 00:00:00 2001 From: Andrey Date: Tue, 27 Nov 2018 16:54:29 +0300 Subject: [PATCH] https://github.com/tensorflow/models/issues/4780 --- research/object_detection/model_lib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/research/object_detection/model_lib.py b/research/object_detection/model_lib.py index b54c6ef2b3d..3b4344b4581 100644 --- a/research/object_detection/model_lib.py +++ b/research/object_detection/model_lib.py @@ -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: