start epoch plus iters gives me the jitters

2 files changed, 1 insertions, 2 deletions

diff --git a/dir-to-movie.pl b/dir-to-movie.pl
index 40fa853..0698241 100755
--- a/dir-to-movie.pl
+++ b/dir-to-movie.pl
@@ -38,7 +38,6 @@ chomp $pwd;
 $pwd .= '/';
 print $pwd . "\n";
 
-
 mkdir('./tmp');
 
 for my $line (@lines) {
diff --git a/train.py b/train.py
index ae3095f..3cf3627 100755
--- a/train.py
+++ b/train.py
@@ -44,7 +44,7 @@ display_delta = total_steps % opt.display_freq
 print_delta = total_steps % opt.print_freq
 save_delta = total_steps % opt.save_latest_freq
 
-for epoch in range(start_epoch, opt.niter + opt.niter_decay + 1):
+for epoch in range(start_epoch, opt.niter + opt.niter_decay + start_epoch):
     epoch_start_time = time.time()
     if epoch != start_epoch:
         epoch_iter = epoch_iter % dataset_size