wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-08 16:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

active windowing

Browse Source
This commit is contained in:
Eren Golge
2019-04-05 08:40:09 +02:00
parent a23674e4dd
commit 043e49f367
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/tacotron2.py
+3 -4
View File
@@ -124,14 +124,13 @@ class Attention(nn.Module):
attention_dim)
self._mask_value = -float("inf")
self.windowing = windowing
if self.windowing:
self.win_back = 1
self.win_front = 3
self.win_idx = None
self.win_idx = None
self.norm = norm
def init_win_idx(self):
self.win_idx = -1
self.win_back = 1
self.win_front = 3
def get_attention(self, query, processed_inputs, attention_cat):
processed_query = self.query_layer(query.unsqueeze(1))