From 11b227edfc79191af05f46eb07e0747e177f0bb7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 4 Aug 2020 10:30:40 -0600 Subject: [PATCH] Whoops --- 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 5d4f4274..11625dbf 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -652,7 +652,7 @@ class SRGANModel(BaseModel): for p in self.netD_grad.parameters(): p.requires_grad = True self.optimizer_D_grad.zero_grad() - for var_ref, fake_H in zip(self.var_ref_skips, self.fake_H): + for var_ref, fake_H in zip(var_ref_skips, self.fake_H): fake_H_grad = self.get_grad_nopadding(fake_H) var_ref_grad = self.get_grad_nopadding(var_ref) pred_d_real_grad = self.netD_grad(var_ref_grad)