From 1a764f00f8fced8696592bee776d2154b1e8e5a1 Mon Sep 17 00:00:00 2001
From: junyanz <junyanz@berkeley.edu>
Date: Fri, 6 Oct 2017 11:18:52 -0700
Subject: fix minor bugs

---
 train.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'train.py')

diff --git a/train.py b/train.py
index 4d80eb6..7d2a5e9 100644
--- a/train.py
+++ b/train.py
@@ -17,6 +17,7 @@ total_steps = 0
 for epoch in range(opt.epoch_count, opt.niter + opt.niter_decay + 1):
     epoch_start_time = time.time()
     epoch_iter = 0
+
     for i, data in enumerate(dataset):
         iter_start_time = time.time()
         total_steps += opt.batchSize
@@ -47,6 +48,4 @@ for epoch in range(opt.epoch_count, opt.niter + opt.niter_decay + 1):
 
     print('End of epoch %d / %d \t Time Taken: %d sec' %
           (epoch, opt.niter + opt.niter_decay, time.time() - epoch_start_time))
-
-    if epoch > opt.niter:
-        model.update_learning_rate()
+    model.update_learning_rate()
-- 
cgit v1.2.3-70-g09d2