From 85ee64b8d9091291f63e101a44505b7f9e9e3b8e Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 27 Jul 2020 15:28:55 -0600 Subject: [PATCH] Turn down feadisc intensity Honestly - this feature is probably going to be removed soon, so backwards compatibility is not a huge deal anymore. --- 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 317ec659..6a75acd0 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -389,7 +389,7 @@ class SRGANModel(BaseModel): l_d_fea_fake = 0 if self.opt['train']['gan_type'] == 'pixgan_fea': # Compute a feature loss which is added to the GAN loss computed later to guide the discriminator better. - disc_fea_scale = .5 + disc_fea_scale = .1 _, fea_real = self.netD(var_ref, output_feature_vector=True) actual_fea = self.netF(var_ref) l_d_fea_real = self.cri_fea(fea_real, actual_fea) * disc_fea_scale / self.mega_batch_factor