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

guided attn #4

This commit is contained in:
Eren Golge 2018-04-24 11:30:58 -07:00
parent 937d1140f4
commit 9d6c08ab72
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -115,7 +115,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
M[n, t] = val M[n, t] = val
e_x = np.exp(M - np.max(M)) e_x = np.exp(M - np.max(M))
M = e_x / e_x.sum(axis=0) # only difference M = e_x / e_x.sum(axis=0) # only difference
M = Variable(M) M = Variable(torch.FloatTensor(M))
M = torch.stack([M]*32) M = torch.stack([M]*32)
# forward pass # forward pass