mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
make attn guiding optional #4
This commit is contained in:
@@ -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