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

guided attn #12

Browse Source
This commit is contained in:
Eren Golge committed 2018-04-24 11:41:39 -07:00
1 parent e87d7d8d26
commit 3161a16101
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
View File
@@ -130,7 +130,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
mel_lengths_var)
print(M.shape)
print(alignments.shape)
attention_loss = criterion(M, alignments, mel_lengths_var)
attention_loss = criterion(alignments, M, mel_lengths_var)
loss = mel_loss + linear_loss + 0.2 * attention_loss
# backpass and check the grad norm