From b1ac67197868f8defd54d4a63960ac2f84bf942f Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 30 Apr 2018 06:12:56 -0700 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3e6e6083..94f7e8ec 100644 --- a/train.py +++ b/train.py @@ -152,7 +152,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, epoch): progbar.update(num_iter+1, values=list(progbar_display.items())) avg_linear_loss += linear_loss.item() avg_mel_loss += mel_loss.item() - avg_stop_loss += st_loss.item() + avg_stop_loss += stop_loss.item() # Plot Training Iter Stats tb.add_scalar('TrainIterLoss/TotalLoss', loss.item(), current_step)