Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2026-02-25 04:31:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

make attn guiding optional #5

Browse Source
This commit is contained in:
Eren Golge
2018-04-25 05:52:41 -07:00
parent 52dc46ea75
commit cd441f69bb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -148,7 +148,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
progbar_display['grad_norm'] = grad_norm
# update
progbar.update(num_iter+1, values=[progbar_display.items()])
progbar.update(num_iter+1, values=[list(progbar_display.items())])
avg_linear_loss += linear_loss.data[0]
avg_mel_loss += mel_loss.data[0]