From a66fbb32b666f240dada9f605447fb605ea2ef40 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 31 Jul 2020 16:59:23 -0600 Subject: [PATCH] Fix fixed_disc DataParallel issue --- 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 cce3f8be..73cdb27d 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -364,7 +364,7 @@ class SRGANModel(BaseModel): l_g_fix_disc = torch.zeros(1, requires_grad=False).squeeze() for fixed_disc in self.fixed_disc_nets: - weight = fixed_disc.fdisc_weight + weight = fixed_disc.module.fdisc_weight real_fea = fixed_disc(pix).detach() fake_fea = fixed_disc(fea_GenOut) l_g_fix_disc += weight * self.cri_fea(fake_fea, real_fea)