mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
make attn guiding optional #4
This commit is contained in:
1 parent
895767ec87
commit
52dc46ea75
1 file changed
+1
-1
@@ -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=[tuple(progbar_display.iteritems())])
|
||||
progbar.update(num_iter+1, values=[progbar_display.items()])
|
||||
avg_linear_loss += linear_loss.data[0]
|
||||
avg_mel_loss += mel_loss.data[0]
|
||||
|
||||
|
||||
Reference in new issue
Block a user