From 375d3c4b9578e85ebae0645f624478e9cdc7bc99 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 25 Apr 2018 05:53:09 -0700 Subject: [PATCH] make attn guiding optional #6 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]