kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix fixed_disc DataParallel issue

Browse Source
This commit is contained in:
James Betker 2020-07-31 16:59:23 -06:00
parent 8dd44182e6
commit a66fbb32b6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/SRGAN_model.py
View File

@ -364,7 +364,7 @@ class SRGANModel(BaseModel):
l_g_fix_disc = torch.zeros(1, requires_grad=False).squeeze() l_g_fix_disc = torch.zeros(1, requires_grad=False).squeeze()
for fixed_disc in self.fixed_disc_nets: 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() real_fea = fixed_disc(pix).detach()
fake_fea = fixed_disc(fea_GenOut) fake_fea = fixed_disc(fea_GenOut)
l_g_fix_disc += weight * self.cri_fea(fake_fea, real_fea) l_g_fix_disc += weight * self.cri_fea(fake_fea, real_fea)