diff options
Diffstat (limited to 'util/visualizer.py')
| -rw-r--r-- | util/visualizer.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index 08eeb75..a98b512 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -56,7 +56,7 @@ class Visualizer(): if idx % ncols == 0: label_html += '<tr>%s</tr>' % label_html_row label_html_row = '' - white_image = np.ones_like(image_numpy.transpose([2, 0, 1]))*255 + white_image = np.ones_like(image_numpy.transpose([2, 0, 1])) * 255 while idx % ncols != 0: images.append(white_image) label_html_row += '<td></td>' @@ -114,8 +114,8 @@ class Visualizer(): win=self.display_id) # errors: same format as |errors| of plotCurrentErrors - def print_current_errors(self, epoch, i, errors, t): - message = '(epoch: %d, iters: %d, time: %.3f) ' % (epoch, i, t) + def print_current_errors(self, epoch, i, errors, t, t_data): + message = '(epoch: %d, iters: %d, time: %.3f, data: %.3f) ' % (epoch, i, t, t_data) for k, v in errors.items(): message += '%s: %.3f ' % (k, v) |