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

batch-wise operation

This commit is contained in:
Eren Golge 2019-05-24 13:40:56 +02:00
parent 3a4a3e571a
commit 2586be7d33
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
layers/tacotron2.py
View File

@ -222,7 +222,11 @@ class Attention(nn.Module):
# force incremental alignment # force incremental alignment
if not self.training: if not self.training:
val, n = prev_alpha.max(1) val, n = prev_alpha.max(1)
alignment[:, n+2 :] = 0 if alignment.shape[0] == 1:
alignment[:, n+2:] = 0
else:
for b in range(alignment.shape[0]):
alignment[b, n[b]+2:]
alpha = (((1 - self.u) * self.alpha.clone().to(inputs.device) + alpha = (((1 - self.u) * self.alpha.clone().to(inputs.device) +
self.u * prev_alpha) + 1e-8) * alignment self.u * prev_alpha) + 1e-8) * alignment
self.alpha = alpha / alpha.sum(dim=1).unsqueeze(1) self.alpha = alpha / alpha.sum(dim=1).unsqueeze(1)