From c41ab7acdb1e9191d91991d6461ddc4dff3243f7 Mon Sep 17 00:00:00 2001
From: Ruotian Luo <rluo@ttic.edu>
Date: Tue, 18 Apr 2017 21:10:01 -0500
Subject: Use visdom instead of display

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train.py')

diff --git a/train.py b/train.py
index e85042f..da497ce 100644
--- a/train.py
+++ b/train.py
@@ -32,7 +32,7 @@ for epoch in range(1, opt.niter + opt.niter_decay + 1):
             errors = model.get_current_errors()
             visualizer.print_current_errors(epoch, epoch_iter, errors, iter_start_time)
             if opt.display_id > 0:
-                visualizer.plot_current_errors(epoch, epoch_iter, opt, errors)
+                visualizer.plot_current_errors(epoch, float(epoch_iter)/num_train, opt, errors)
 
         if total_steps % opt.save_latest_freq == 0:
             print('saving the latest model (epoch %d, total_steps %d)' %
-- 
cgit v1.2.3-70-g09d2