From 464ac9e26fb874ff4d1fea54bcc45ad33dc53b5b Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 16 Jun 2018 21:17:06 +0200 Subject: get --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 8fef794..2a29c11 100644 --- a/train.py +++ b/train.py @@ -54,5 +54,5 @@ if __name__ == '__main__': model.save(opt.epoch_count + epoch) print('End of epoch %d / %d \t Time Taken: %d sec' % - (opt.epoch_count + epoch, opt.niter + opt.niter_decay, time.time() - epoch_start_time)) + (opt.epoch_count + epoch, opt.epoch_count + opt.niter + opt.niter_decay, time.time() - epoch_start_time)) model.update_learning_rate() -- cgit v1.2.3-70-g09d2