mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Whoops

Browse Source
This commit is contained in:
James Betker 2020-08-04 10:30:40 -06:00
parent 6d25bcd5df
commit 11b227edfc
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

@ -652,7 +652,7 @@ class SRGANModel(BaseModel):
for p in self.netD_grad.parameters(): for p in self.netD_grad.parameters():
p.requires_grad = True p.requires_grad = True
self.optimizer_D_grad.zero_grad() 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) fake_H_grad = self.get_grad_nopadding(fake_H)
var_ref_grad = self.get_grad_nopadding(var_ref) var_ref_grad = self.get_grad_nopadding(var_ref)
pred_d_real_grad = self.netD_grad(var_ref_grad) pred_d_real_grad = self.netD_grad(var_ref_grad)