From af22bed149eb202331b2da403d4b1608923faf93 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 16 Jan 2019 15:53:24 +0100 Subject: [PATCH] set bias --- layers/attention.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/layers/attention.py b/layers/attention.py index ecb38587..fc50a9dd 100644 --- a/layers/attention.py +++ b/layers/attention.py @@ -52,10 +52,10 @@ class LocationSensitiveAttention(nn.Module): stride=1, padding=0, bias=False)) - 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=False) + self.loc_linear = nn.Linear(filters, attn_dim, bias=False) + self.query_layer = nn.Linear(query_dim, attn_dim, bias=False) + self.annot_layer = nn.Linear(annot_dim, attn_dim, bias=False) + self.v = nn.Linear(attn_dim, 1, bias=True) self.processed_annots = None # self.init_layers()