From 032af2eb9430de6d37b207fbd8090c81b6d04479 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 22:43:51 -0400 Subject: [PATCH] next --- stargan/solver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stargan/solver.py b/stargan/solver.py index b7a7798..e460d87 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -727,8 +727,8 @@ class Solver(object): # Save the translated images. x_concat = torch.cat(x_fake_list, dim=3) - result_path = os.path.join(self.result_dir, '{}-images.jpg'.format(i+1)) - # result_path = os.path.join(self.result_dir, '{}-{}-images.jpg'.format(layer_num_orig, i+1)) + # result_path = os.path.join(self.result_dir, '{}-images.jpg'.format(i+1)) + result_path = os.path.join(self.result_dir, '{}-{}-images.jpg'.format(layer_num_orig, i+1)) save_image(self.denorm(x_concat.data.cpu()), result_path, nrow=1, padding=0) # print('Saved real and fake images into {}...'.format(result_path)) if i == 3: