Commit 970f4a7e authored by cclauss's avatar cclauss
Browse files

tcn: Fix undefined 'v' in Python 3

parent 331e1372
...@@ -165,7 +165,7 @@ def parse_sequence_example(serialized_example, num_views): ...@@ -165,7 +165,7 @@ def parse_sequence_example(serialized_example, num_views):
views = tf.stack([sequence_parse[v] for v in view_names]) views = tf.stack([sequence_parse[v] for v in view_names])
lens = [sequence_parse[v].get_shape().as_list()[0] for v in view_names] lens = [sequence_parse[v].get_shape().as_list()[0] for v in view_names]
assert len(set(lens)) == 1 assert len(set(lens)) == 1
seq_len = tf.shape(sequence_parse[v])[0] seq_len = tf.shape(sequence_parse[view_names[-1]])[0]
return context_parse, views, seq_len return context_parse, views, seq_len
......
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