From 59af4ea5a1084db67c5470e2a38304bd9e5c0760 Mon Sep 17 00:00:00 2001
From: "jules@lens" <julescarbon@gmail.com>
Date: Mon, 24 Sep 2018 15:03:10 +0200
Subject: training

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

(limited to 'train.py')

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
-- 
cgit v1.2.3-70-g09d2