From 55a087ac9f39795b3dec1361c4548c4c09c943f9 Mon Sep 17 00:00:00 2001
From: Chintan Shah <chnsh24@gmail.com>
Date: Sun, 6 Oct 2019 14:40:53 -0400
Subject: [PATCH] logging and refactor

---
 model/pytorch/dcrnn_supervisor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py
index f1692be..6a8462a 100644
--- a/model/pytorch/dcrnn_supervisor.py
+++ b/model/pytorch/dcrnn_supervisor.py
@@ -155,9 +155,9 @@ class DCRNNSupervisor:
                 torch.nn.utils.clip_grad_norm_(self.dcrnn_model.parameters(), self.max_grad_norm)
 
                 optimizer.step()
-
+            self._logger.info("epoch complete")
             lr_scheduler.step()
-
+            self._logger.info("evaluating now!")
             val_loss = self.evaluate(dataset='val')
             end_time = time.time()
             if epoch_num % log_every == 0: