diff options
Diffstat (limited to 'util')
| -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( |