From 8bc6778456c86a8b9d88362efb195635d2a0dac3 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 26 Aug 2017 18:28:29 -0700 Subject: add epoch_count --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 0834d37..4d80eb6 100644 --- a/train.py +++ b/train.py @@ -12,15 +12,15 @@ print('#training images = %d' % dataset_size) model = create_model(opt) visualizer = Visualizer(opt) - total_steps = 0 -for epoch in range(1, opt.niter + opt.niter_decay + 1): +for epoch in range(opt.epoch_count, opt.niter + opt.niter_decay + 1): epoch_start_time = time.time() + epoch_iter = 0 for i, data in enumerate(dataset): iter_start_time = time.time() total_steps += opt.batchSize - epoch_iter = total_steps - dataset_size * (epoch - 1) + epoch_iter += opt.batchSize model.set_input(data) model.optimize_parameters() -- cgit v1.2.3-70-g09d2