load iters

1 files changed, 1 insertions, 1 deletions

diff --git a/augment.py b/augment.py
index 6741f04..cb75880 100755
--- a/augment.py
+++ b/augment.py
@@ -74,7 +74,7 @@ if opt.which_epoch == 'latest':
       current_epoch, epoch_iter = np.loadtxt(iter_path , delimiter=',', dtype=int)
     except:
       current_epoch, epoch_iter = 1, 0
-    print('Resuming from epoch %d at iteration %d' % (current_epoch, epoch_iter))        
+    print('Loading epoch %d' % (current_epoch,))        
   else:    
     current_epoch, epoch_iter = 1, 0
 else: