Skip to content

GitLab

Commit 8f16def1 authored by Byeongjoo Ahn's avatar Byeongjoo Ahn Committed by GitHub
Browse files

Fix compatibility for Python3 

iteritems() -> items()
parent 56765e5b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
object_detection/utils/variables_helper.py
View file @ 8f16def1
......@@ -122,7 +122,7 @@ def get_variables_available_in_checkpoint(variables, checkpoint_path):
ckpt_reader = tf.train.NewCheckpointReader(checkpoint_path)
ckpt_vars = ckpt_reader.get_variable_to_shape_map().keys()
vars_in_ckpt = {}
for variable_name, variable in sorted(variable_names_map.iteritems()):
for variable_name, variable in sorted(variable_names_map.items()):
if variable_name in ckpt_vars:
vars_in_ckpt[variable_name] = variable
else:
......
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