From 9c8752432b3f5b0a9f49197d1fe869b512e44f2f Mon Sep 17 00:00:00 2001
From: "jules@lens" <julescarbon@gmail.com>
Date: Fri, 21 Sep 2018 22:53:54 +0200
Subject: epoch problem paranoia

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

(limited to 'augment.py')

diff --git a/augment.py b/augment.py
index cb75880..b5894e1 100755
--- a/augment.py
+++ b/augment.py
@@ -72,6 +72,7 @@ if opt.which_epoch == 'latest':
   if os.path.exists(iter_path):
     try:
       current_epoch, epoch_iter = np.loadtxt(iter_path , delimiter=',', dtype=int)
+      current_epoch = int(current_epoch) - 1
     except:
       current_epoch, epoch_iter = 1, 0
     print('Loading epoch %d' % (current_epoch,))        
-- 
cgit v1.2.3-70-g09d2