diff options
| -rw-r--r-- | util/visualizer.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index 46348be..a718cfc 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -21,7 +21,10 @@ class Visualizer(): self.img_dir = os.path.join(self.web_dir, 'images') print('create web directory %s...' % self.web_dir) util.mkdirs([self.web_dir, self.img_dir]) - + self.log_name = os.path.join(opt.checkpoints_dir, opt.name, 'loss_log.txt') + with open(self.log_name, "a") as log_file: + now = time.strftime("%c") + log_file.write('================ Training Loss (%s) ================\n' % now) # |visuals|: dictionary of images to display or save def display_current_results(self, visuals, epoch): @@ -76,6 +79,8 @@ class Visualizer(): message += '%s: %.3f ' % (k, v) print(message) + with open(self.log_name, "a") as log_file: + log_file.write('%s\n' % message) # save image to the disk def save_images(self, webpage, visuals, image_path): |