From 3c0a2d6efec648dfe208bfd8cfabd105b89c82f3 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 4 Aug 2020 16:43:43 -0600 Subject: [PATCH] Fix grad branch debug out --- 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 433a2a2a..f2d3d2dd 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -404,7 +404,7 @@ class SRGANModel(BaseModel): # Get image gradients for later use. fake_H_grad = self.get_grad_nopadding(fake_GenOut) var_H_grad_nopadding = self.get_grad_nopadding(var_H) - self.spsr_grad_GenOut.append(grad_LR) + self.spsr_grad_GenOut.append(fake_H_branch) else: if random.random() > self.gan_lq_img_use_prob: fea_GenOut, fake_GenOut = self.netG(var_L)