1 files changed, 2 insertions, 1 deletions
diff --git a/train.py b/train.py
index ae129b9..12c9dbc 100644
--- a/train.py
+++ b/train.py
@@ -30,7 +30,8 @@ for epoch in range(1, opt.niter + opt.niter_decay + 1):
 
         if total_steps % opt.print_freq == 0:
             errors = model.get_current_errors()
-            visualizer.print_current_errors(epoch, epoch_iter, errors, iter_start_time)
+            t = (time.time() - iter_start_time) / opt.batchSize
+            visualizer.print_current_errors(epoch, epoch_iter, errors, t)
             if opt.display_id > 0:
                 visualizer.plot_current_errors(epoch, float(epoch_iter)/dataset_size, opt, errors)