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

Fix grad branch debug out

Browse Source
This commit is contained in:
James Betker 2020-08-04 16:43:43 -06:00
parent ec2a795d53
commit 3c0a2d6efe
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

@ -404,7 +404,7 @@ class SRGANModel(BaseModel):
# Get image gradients for later use.
fake_H_grad = self.get_grad_nopadding(fake_GenOut)
var_H_grad_nopadding = self.get_grad_nopadding(var_H)
self.spsr_grad_GenOut.append(grad_LR)
self.spsr_grad_GenOut.append(fake_H_branch)
else:
if random.random() > self.gan_lq_img_use_prob:
fea_GenOut, fake_GenOut = self.netG(var_L)