get

1 files changed, 1 insertions, 1 deletions

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()