larger attention filter size and mode filters

This commit is contained in:
Eren 2018-09-06 15:27:15 +02:00
parent 56c6d0cac8
commit f377cd3cb8
1 changed files with 2 additions and 2 deletions

View File

@ -37,8 +37,8 @@ class LocationSensitiveAttention(nn.Module):
annot_dim, annot_dim,
query_dim, query_dim,
attn_dim, attn_dim,
kernel_size=7, kernel_size=31,
filters=20): filters=32):
super(LocationSensitiveAttention, self).__init__() super(LocationSensitiveAttention, self).__init__()
self.kernel_size = kernel_size self.kernel_size = kernel_size
self.filters = filters self.filters = filters