Skip to content

Commit 35d6291

Browse files
Sindy LoeweSindy Loewe
Sindy Loewe
authored and
Sindy Loewe
committed
unmasking the horizontal stack
1 parent 055dab6 commit 35d6291

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@ def __init__(self, X, conf, full_horizontal=True, h=None):
3131
v_stack_in = v_stack
3232

3333
with tf.variable_scope("v_stack_1"+i):
34-
v_stack_1 = GatedCNN([1, 1, conf.f_map], v_stack_in, False, gated=False, mask=mask).output()
34+
v_stack_1 = GatedCNN([1, 1, conf.f_map], v_stack_in, False, gated=False, mask=None).output()
3535

3636
with tf.variable_scope("h_stack"+i):
3737
h_stack = GatedCNN([filter_size if full_horizontal else 1, filter_size, conf.f_map], h_stack_in, True, payload=v_stack_1, mask=mask, conditional=self.h).output()
3838

3939
with tf.variable_scope("h_stack_1"+i):
40-
h_stack_1 = GatedCNN([1, 1, conf.f_map], h_stack, True, gated=False, mask=mask).output()
40+
h_stack_1 = GatedCNN([1, 1, conf.f_map], h_stack, True, gated=False, mask=None).output()
4141
if residual:
4242
h_stack_1 += h_stack_in # Residual connection
4343
h_stack_in = h_stack_1

0 commit comments

Comments
 (0)