coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

larger attn loss

pull/10/head
Eren Golge 2018-04-24 13:21:10 -07:00
parent 85bbab74b3
commit 17ef3de1b9
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

@ -129,7 +129,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
linear_spec_var[:, :, :n_priority_freq],
mel_lengths_var)
attention_loss = criterion(alignments, M, mel_lengths_var)
loss = mel_loss + linear_loss + 0.2 * attention_loss
loss = mel_loss + linear_loss + attention_loss
# backpass and check the grad norm
loss.backward()