From 34c9bbea23a1a61eaf866ae95f439e29d22ce242 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 22 Jun 2018 20:37:42 +0200 Subject: train --- train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'train.py') diff --git a/train.py b/train.py index 3cf3627..0fe892d 100755 --- a/train.py +++ b/train.py @@ -44,7 +44,9 @@ 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 + start_epoch): +print("{} {}".format(start_epoch, start_epoch + opt.niter + opt.niter_decay + 1)) + +for epoch in range(start_epoch, start_epoch + opt.niter + opt.niter_decay + 1): epoch_start_time = time.time() if epoch != start_epoch: epoch_iter = epoch_iter % dataset_size -- cgit v1.2.3-70-g09d2