1 files changed, 1 insertions, 0 deletions
diff --git a/train.py b/train.py
index 733282a..5aacfb9 100755
--- a/train.py
+++ b/train.py
@@ -52,6 +52,7 @@ for epoch in range(start_epoch, start_epoch + opt.niter):
     # if epoch != start_epoch:
     #     epoch_iter = epoch_iter % dataset_size 
     epoch_iter = 0
+    np.savetxt(iter_path, (int(epoch)+1, epoch_iter), delimiter=',', fmt='%d')
     for i, data in enumerate(dataset): #, start=epoch_iter
         iter_start_time = time.time()
         total_steps += opt.batchSize