smaller windowing range

2026-09-09 11:16:00 +08:00 · 2019-04-02 10:35:13 +02:00
parent 68f8ef730d
commit 6e8b66aa89
1 changed files with 2 additions and 2 deletions
@@ -125,8 +125,8 @@ class Attention(nn.Module):
        self._mask_value = -float("inf")
        self.windowing = windowing
        if self.windowing:
-            self.win_back = 3
-            self.win_front = 6
+            self.win_back = 1
+            self.win_front = 3
            self.win_idx = None
        self.norm = norm