From ba08e98bd9a155fe677a9ff3e3285cf0299c53d0 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Tue, 14 May 2019 21:33:37 +0200
Subject: [PATCH] revert debug changes

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

diff --git a/train.py b/train.py
index ae6e8990..b3eb4f39 100644
--- a/train.py
+++ b/train.py
@@ -465,10 +465,9 @@ def main(args):
         best_loss = float('inf')
 
     for epoch in range(0, c.epochs):
-        # train_loss, current_step = train(model, criterion, criterion_st,
-        #                                  optimizer, optimizer_st, scheduler,
-        #                                  ap, epoch)
-        train_loss, current_step = 0, 0
+        train_loss, current_step = train(model, criterion, criterion_st,
+                                         optimizer, optimizer_st, scheduler,
+                                         ap, epoch)
         val_loss = evaluate(model, criterion, criterion_st, ap, current_step, epoch)
         print(
             " | > Training Loss: {:.5f}   Validation Loss: {:.5f}".format(