From ec1b8d950bb489f1e04c43aefe6fa5e51a7d89fa Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 21 Jun 2018 13:11:11 +0200 Subject: start epoch plus iters gives me the jitters --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train.py') 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 -- cgit v1.2.3-70-g09d2