Explicit padding for unbalanced padding sizes

layers/attention.py

@@ -43,13 +43,15 @@ class LocationSensitiveAttention(nn.Module):
         self.kernel_size = kernel_size
         self.filters = filters
         padding = [(kernel_size - 1) // 2, (kernel_size - 1) // 2]
-        self.loc_conv = nn.Conv1d(
+        self.loc_conv = nn.Sequential(
+            nn.ConstantPad1d(padding, 0),
+            nn.Conv1d(
                 2,
                 filters,
                 kernel_size=kernel_size,
                 stride=1,
-            padding=padding,
+                padding=0,
-            bias=False)
+                bias=False))
         self.loc_linear = nn.Linear(filters, attn_dim)
         self.query_layer = nn.Linear(query_dim, attn_dim, bias=True)
         self.annot_layer = nn.Linear(annot_dim, attn_dim, bias=True)
@@ -100,8 +102,8 @@ class AttentionRNNCell(nn.Module):
                 annot_dim, rnn_dim, out_dim)
         else:
             raise RuntimeError(" Wrong alignment model name: {}. Use\
-                'b' (Bahdanau) or 'ls' (Location Sensitive)."
+                'b' (Bahdanau) or 'ls' (Location Sensitive).".format(
-                               .format(align_model))
+                align_model))
 
     def forward(self, memory, context, rnn_state, annots, atten, mask):
         """