diff options
Diffstat (limited to 'util/visualizer.py')
| -rwxr-xr-x | util/visualizer.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index f41c55a..c11904d 100755 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -1,4 +1,4 @@ -### Copyright (C) 2017 NVIDIA Corporation. All rights reserved. +### Copyright (C) 2017 NVIDIA Corporation. All rights reserved. ### Licensed under the CC BY-NC-SA 4.0 license (https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode). import numpy as np import os @@ -6,7 +6,7 @@ import ntpath import time from . import util from . import html -import scipy.misc +import scipy.misc try: from StringIO import StringIO # Python 2.7 except ImportError: @@ -66,7 +66,7 @@ class Visualizer(): util.save_image(image_numpy, img_path) # update website - webpage = html.HTML(self.web_dir, 'Experiment name = %s' % self.name, reflesh=1) + webpage = html.HTML(self.web_dir, 'Experiment name = %s' % self.name, refresh=5) for n in range(epoch, 0, -1): webpage.add_header('epoch [%d]' % n) ims = [] @@ -96,7 +96,7 @@ class Visualizer(): # errors: dictionary of error labels and values def plot_current_errors(self, errors, step): if self.tf_log: - for tag, value in errors.items(): + for tag, value in errors.items(): summary = self.tf.Summary(value=[self.tf.Summary.Value(tag=tag, simple_value=value)]) self.writer.add_summary(summary, step) |