diff options
Diffstat (limited to 'util/visualizer.py')
| -rw-r--r-- | util/visualizer.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |