From 6d0635e8c12bcd450fe8b65a40cecb917955b7bd Mon Sep 17 00:00:00 2001
From: Rafael Valle <rafaelvalle@ngvpn01-173-33.dyn.scz.us.nvidia.com>
Date: Thu, 9 Jan 2020 01:37:57 -0800
Subject: [PATCH] train.py: printing correct variable

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

diff --git a/train.py b/train.py
index 61f0888..88b0949 100644
--- a/train.py
+++ b/train.py
@@ -142,7 +142,7 @@ def validate(model, criterion, valset, iteration, batch_size, n_gpus,
 
     model.train()
     if rank == 0:
-        print("Validation loss {}: {:9f}  ".format(iteration, reduced_val_loss))
+        print("Validation loss {}: {:9f}  ".format(iteration, val_loss))
         logger.log_validation(val_loss, model, y, y_pred, iteration)