larger attention filter size and mode filters

2026-09-09 11:16:00 +08:00 · 2018-09-06 15:27:15 +02:00
parent 56c6d0cac8
commit f377cd3cb8
1 changed files with 2 additions and 2 deletions
@@ -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