diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index 67eea696..317ec659 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -235,7 +235,6 @@ class SRGANModel(BaseModel): else: # If not provided, use provided LQ for anyplace where the GAN would have been used. self.gan_img = self.var_L - self.gan_lq_img_use_prob = 0 # Safety valve for not goofing. if not self.updated: self.netG.module.update_model(self.optimizer_G, self.schedulers[0])