larger attention filter size and mode filters

2018-09-06 15:27:15 +02:00 · 2018-09-06 15:27:15 +02:00 · f377cd3cb8
parent 56c6d0cac8
commit f377cd3cb8
1 changed files with 2 additions and 2 deletions
--- a/layers/attention.py
+++ b/layers/attention.py
@ -37,8 +37,8 @@ class LocationSensitiveAttention(nn.Module):
                 annot_dim,
                 query_dim,
                 attn_dim,
-                 kernel_size=7,
-                 filters=20):
+                 kernel_size=31,
+                 filters=32):
        super(LocationSensitiveAttention, self).__init__()
        self.kernel_size = kernel_size
        self.filters = filters