deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug reporting spsr gan weight

Browse Source
This commit is contained in:
James Betker 2020-08-25 16:37:45 -06:00
parent 96586d6592
commit 1fe16f71dd
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

@ -840,7 +840,7 @@ class SRGANModel(BaseModel):
self.add_log_entry('l_d_fake', l_d_fake_log.detach().item())
self.add_log_entry('D_fake', torch.mean(pred_d_fake.detach()))
self.add_log_entry('D_diff', torch.mean(pred_d_fake.detach()) - torch.mean(pred_d_real.detach()))
if self.spsr_enabled:
if self.spsr_enabled and self.l_gan_grad_w > 0:
self.add_log_entry('l_d_real_grad', l_d_real_grad.detach().item())
self.add_log_entry('l_d_fake_grad', l_d_fake_grad.detach().item())
self.add_log_entry('D_fake_grad', torch.mean(pred_d_fake_grad.detach()))