diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:04:17 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:04:17 -0700 |
| commit | 9cecc5fc9e19e08bf4c0d91b64fe9d45c83c610c (patch) | |
| tree | 9332a8e5d560c20a3502b9061607f3ea55e008da | |
| parent | 07274cd211e6fe180b8287eb65519e66f87d3e2a (diff) | |
fixed a small bug in plotting error for python 3
| -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 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( |