From 9cecc5fc9e19e08bf4c0d91b64fe9d45c83c610c Mon Sep 17 00:00:00 2001
From: Taesung Park <taesung_park@berkeley.edu>
Date: Thu, 20 Apr 2017 03:04:17 -0700
Subject: fixed a small bug in plotting error for python 3

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

(limited to 'util')

diff --git a/util/visualizer.py b/util/visualizer.py
index 7df9cce..a839896 100644
--- a/util/visualizer.py
+++ b/util/visualizer.py
@@ -55,7 +55,7 @@ class Visualizer():
     # errors: dictionary of error labels and values
     def plot_current_errors(self, epoch, counter_ratio, opt, errors):
         if not hasattr(self, 'plot_data'):
-            self.plot_data = {'X':[],'Y':[], 'legend':errors.keys()}
+            self.plot_data = {'X':[],'Y':[], 'legend':list(errors.keys())}
         self.plot_data['X'].append(epoch + counter_ratio)
         self.plot_data['Y'].append([errors[k] for k in self.plot_data['legend']])
         self.vis.line(
-- 
cgit v1.2.3-70-g09d2