From 3b72a659c38141e502b74bee65ca08d51dc3eabf Mon Sep 17 00:00:00 2001 From: junyanz Date: Mon, 12 Jun 2017 19:23:18 -0700 Subject: save training losses to the file (log_file.txt) --- util/visualizer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'util/visualizer.py') 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): -- cgit v1.2.3-70-g09d2