From 9245992817195008061e3970828090377857182e Mon Sep 17 00:00:00 2001
From: Eren G <egolge@mozilla.com>
Date: Mon, 9 Jul 2018 15:56:30 +0200
Subject: [PATCH] Print step information in train

---
 train.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 34f96829..01ed1bd4 100644
--- a/train.py
+++ b/train.py
@@ -150,9 +150,10 @@ def train(model, criterion, criterion_st, data_loader, optimizer, optimizer_st,
         #                                    ('grad_norm_st', grad_norm_st.item())])
 
         if current_step % c.print_step == 0:
-            print(" | | > TotalLoss: {:.5f}\t LinearLoss: {:.5f}\t MelLoss: \
-                  {:.5f}\t StopLoss: {:.5f}\t GradNorm: {:.5f}\t \
-                  GradNormST: {:.5f}".format(loss.item(),
+            print(" | | > Step:{}\tGlobalStep:{}\tTotalLoss:{:.5f}\tLinearLoss:{:.5f}\tMelLoss:\
+                  {:.5f}\tStopLoss:{:.5f}\tGradNorm:{:.5f}\t\
+                  GradNormST: {:.5f}".format(num_iter, current_step,
+                                             loss.item(),
                                              linear_loss.item(),
                                              mel_loss.item(),
                                              stop_loss.item(),