dynamic pooling typo

805094591 · Dec 6, 2018 · 97898f6 · 97898f6
1 parent 60b698a
commit 97898f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/simnet/train/tf/layers/tf_layers.py b/tools/simnet/train/tf/layers/tf_layers.py
@@ -505,7 +505,7 @@ def __init__(self, seq_len1, seq_len2, win_size1, win_size2,
         self.dpool_size1 = dpool_size1
         self.dpool_size2 = dpool_size2
         self.kernel_size = kernel_size
-        filter_shape = [self.seq_len1, self.seq_len2, 1, self.kernel_size]
+        filter_shape = [self.win_size1, self.win_size2, 1, self.kernel_size]
         self.conv_w = tf.Variable(tf.truncated_normal(filter_shape, stddev=0.1),
                                   name="conv")
         self.bias = tf.Variable(tf.constant(0.1, shape=[self.kernel_size]),