Commit 1dbdbbb6 authored by DefineFC's avatar DefineFC
Browse files

py3 compatibility

parent 2c4762d0
......@@ -436,7 +436,7 @@ def create_model_fn(detection_model_fn, configs, hparams, use_tpu=False):
# 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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment