Attention bias setting Revert to old

This commit is contained in:
Eren Golge 2019-02-06 16:23:01 +01:00
parent eb839a7acd
commit d28bbe09fb
1 changed files with 1 additions and 1 deletions

View File

@ -55,7 +55,7 @@ class LocationSensitiveAttention(nn.Module):
self.loc_linear = nn.Linear(filters, attn_dim, bias=True)
self.query_layer = nn.Linear(query_dim, attn_dim, bias=True)
self.annot_layer = nn.Linear(annot_dim, attn_dim, bias=True)
self.v = nn.Linear(attn_dim, 1, bias=True)
self.v = nn.Linear(attn_dim, 1, bias=False)
self.processed_annots = None
# self.init_layers()