Change window size for attention

2019-02-18 13:06:26 +01:00 · 2019-02-18 13:06:26 +01:00 · 1c99be2ffd
parent 97a16cedbf
commit 1c99be2ffd
1 changed files with 2 additions and 2 deletions
--- a/layers/attention.py
+++ b/layers/attention.py
@ -118,8 +118,8 @@ class AttentionRNNCell(nn.Module):
        self.rnn_cell = nn.GRUCell(annot_dim + memory_dim, rnn_dim)
        self.windowing = windowing
        if self.windowing:
-            self.win_back = 1
-            self.win_front = 3
+            self.win_back = 3
+            self.win_front = 6
            self.win_idx = None
        # pick bahdanau or location sensitive attention
        if align_model == 'b':