Unverified Commit 42b3ab23 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #3272 from cclauss/undefined-v-in-python3

tcn: Fix undefined 'v' in Python 3
parents 9b401346 70629555
......@@ -165,7 +165,7 @@ def parse_sequence_example(serialized_example, num_views):
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]
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
......
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