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