From efef2906b0a3f4fd265823ff4b4b99ccebeb6d05 Mon Sep 17 00:00:00 2001
From: junyanz <junyanz@berkeley.edu>
Date: Tue, 18 Apr 2017 03:56:29 -0700
Subject: add unet and update README

---
 util/visualizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'util/visualizer.py')

diff --git a/util/visualizer.py b/util/visualizer.py
index 0b8578e..4daf506 100644
--- a/util/visualizer.py
+++ b/util/visualizer.py
@@ -27,7 +27,7 @@ class Visualizer():
     def display_current_results(self, visuals, epoch):
         if self.display_id > 0: # show images in the browser
             idx = 0
-            for label, image_numpy in visuals:
+            for label, image_numpy in visuals.items():
                 image_numpy = np.flipud(image_numpy)
                 self.display.image(image_numpy, title=label,
                                    win=self.display_id + idx)
-- 
cgit v1.2.3-70-g09d2