Skip to content

GitLab

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

tcn: Fix undefined 'v' in Python 3

parent 331e1372
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/tcn/data_providers.py
View file @ 970f4a7e
...@@ -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