From 35d6291cf4be1400ac7033d30cbfde66fbd6572b Mon Sep 17 00:00:00 2001 From: Sindy Loewe Date: Mon, 18 Jun 2018 15:12:43 +0200 Subject: [PATCH] unmasking the horizontal stack --- models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index 6f7df4a..b018666 100644 --- a/models.py +++ b/models.py @@ -31,13 +31,13 @@ def __init__(self, X, conf, full_horizontal=True, h=None): v_stack_in = v_stack with tf.variable_scope("v_stack_1"+i): - v_stack_1 = GatedCNN([1, 1, conf.f_map], v_stack_in, False, gated=False, mask=mask).output() + v_stack_1 = GatedCNN([1, 1, conf.f_map], v_stack_in, False, gated=False, mask=None).output() with tf.variable_scope("h_stack"+i): 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() with tf.variable_scope("h_stack_1"+i): - h_stack_1 = GatedCNN([1, 1, conf.f_map], h_stack, True, gated=False, mask=mask).output() + h_stack_1 = GatedCNN([1, 1, conf.f_map], h_stack, True, gated=False, mask=None).output() if residual: h_stack_1 += h_stack_in # Residual connection h_stack_in = h_stack_1