Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

active windowing

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
layers/tacotron2.py
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))