This commit is contained in:
Eren Golge 2019-04-02 10:35:13 +02:00
Родитель 68f8ef730d
Коммит 6e8b66aa89
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

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