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

bug fix #2

Browse Source
This commit is contained in:
Eren Golge
2019-04-18 18:55:37 +02:00
parent 9ba13b2d2f
commit 38213dffe9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/tacotron2.py
+1 -1
View File
@@ -183,7 +183,7 @@ class Attention(nn.Module):
def apply_forward_attention(self, inputs, alignment, processed_query):
# forward attention
prev_alpha = F.pad(self.alpha[:, :-1].clone(), (1, 0, 0, 0)).to(inputs.device)
alpha = (((1-self.u) * self.alpha.clone().to(inputs.device) + self.u * prev_alpha)) * alignment
alpha = (((1-self.u) * self.alpha.clone().to(inputs.device) + self.u * prev_alpha) + 1e-8) * alignment
self.alpha = alpha / alpha.sum(dim=1).unsqueeze(1)
# compute context
context = torch.bmm(self.alpha.unsqueeze(1), inputs)