make attn guiding optional #4

This commit is contained in:
Eren Golge 2018-04-25 05:51:59 -07:00
parent 3c8ded5a18
commit 9cf1df9219
1 changed files with 1 additions and 1 deletions

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=[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]