From 12da993da89e7c6638d279e15b0d7ceb697425dc Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 8 Jul 2020 22:07:09 -0600 Subject: [PATCH] More fixes... --- codes/models/SRGAN_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index 427b027c..d7987664 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -435,7 +435,7 @@ class SRGANModel(BaseModel): utils.save_image(self.pix[i].cpu(), os.path.join(sample_save_path, "pix", "%05i_%02i.png" % (step, i))) if multi_gen: utils.save_image(self.fake_GenOut[i][0].cpu(), os.path.join(sample_save_path, "gen", "%05i_%02i.png" % (step, i))) - if step % self.D_update_ratio == 0 and step > self.D_init_iters: + if self.l_gan_w > 0 and step > self.G_warmup: utils.save_image(var_ref_skips[i].cpu(), os.path.join(sample_save_path, "ref", "%05i_%02i.png" % (step, i))) utils.save_image(self.fake_H[i], os.path.join(sample_save_path, "disc_fake", "%05i_%02i.png" % (step, i))) else: