From dcdbdc8f55999b57d9052673345e15508db6265f Mon Sep 17 00:00:00 2001
From: Eren G <egolge@mozilla.com>
Date: Thu, 19 Jul 2018 16:55:00 +0200
Subject: [PATCH] Fix for eval time logging

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 26f0c280..92d217f7 100644
--- a/train.py
+++ b/train.py
@@ -283,7 +283,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
             #                                    ('linear_loss', linear_loss.item()),
             #                                    ('mel_loss', mel_loss.item()),
             #                                    ('stop_loss', stop_loss.item())])
-            if current_step % c.print_step == 0:
+            if num_iter % c.print_step == 0:
                 print(" | | > TotalLoss: {:.5f}   LinearLoss: {:.5f}   MelLoss:{:.5f}  "\
                       "StopLoss: {:.5f}  ".format(loss.item(),
                                                           linear_loss.item(),