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

Merge pull request #404 from Edresson/patch-3 

Fix bug in Graves Attn
This commit is contained in:
Eren Gölge 2020-05-07 17:37:25 +02:00 committed by GitHub
parent 7a9247f91f cce13ee245
commit 7292d303b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/common_layers.py
View File

@ -138,7 +138,7 @@ class GravesAttention(nn.Module):
def init_states(self, inputs): def init_states(self, inputs):
if self.J is None or inputs.shape[1]+1 > self.J.shape[-1]: if self.J is None or inputs.shape[1]+1 > self.J.shape[-1]:
self.J = torch.arange(0, inputs.shape[1]+2).to(inputs.device) + 0.5 self.J = torch.arange(0, inputs.shape[1]+2.0).to(inputs.device) + 0.5
self.attention_weights = torch.zeros(inputs.shape[0], inputs.shape[1]).to(inputs.device) self.attention_weights = torch.zeros(inputs.shape[0], inputs.shape[1]).to(inputs.device)
self.mu_prev = torch.zeros(inputs.shape[0], self.K).to(inputs.device) self.mu_prev = torch.zeros(inputs.shape[0], self.K).to(inputs.device)