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)