From 79bc4d248b97ccc70e5170f964a2cf585a2d9311 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Fri, 14 Sep 2018 17:57:26 +0200
Subject: printing

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

(limited to 'train.py')

diff --git a/train.py b/train.py
index be23dd6..15f0e37 100755
--- a/train.py
+++ b/train.py
@@ -50,6 +50,7 @@ for epoch in range(start_epoch, start_epoch + opt.niter):
     epoch_start_time = time.time()
     # if epoch != start_epoch:
     #     epoch_iter = epoch_iter % dataset_size 
+    epoch_iter = 0
     for i, data in enumerate(dataset): #, start=epoch_iter
         iter_start_time = time.time()
         total_steps += opt.batchSize
-- 
cgit v1.2.3-70-g09d2