1 files changed, 1 insertions, 0 deletions
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