diff --git a/train.py b/train.py index 7f615dbd..02b2476a 100644 --- a/train.py +++ b/train.py @@ -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=[list(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]