smaller windowing range

pull/10/head
Eren Golge 2019-04-02 10:35:13 +02:00
parent 68f8ef730d
commit 6e8b66aa89
1 changed files with 2 additions and 2 deletions

View File

@ -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