Fix pickling error

See tensorflow#4780
guildai · Jul 30, 2018 · d88f1fb · d88f1fb
1 parent a26c3d6
commit d88f1fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/research/object_detection/model_lib.py b/research/object_detection/model_lib.py
@@ -378,7 +378,7 @@ def tpu_scaffold():
         eval_metrics = ['coco_detection_metrics']
       eval_metric_ops = eval_util.get_eval_metric_ops_for_evaluators(
           eval_metrics,
-          category_index.values(),
+          list(category_index.values()),
           eval_dict,
           include_metrics_per_category=eval_config.include_metrics_per_category)
       for loss_key, loss_tensor in iter(losses_dict.items()):