forked from mrq/DL-Art-School
ffffffffffffffffff
This commit is contained in:
parent
1fe16f71dd
commit
8bae0de769
|
@ -840,7 +840,7 @@ class SRGANModel(BaseModel):
|
||||||
self.add_log_entry('l_d_fake', l_d_fake_log.detach().item())
|
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_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()))
|
self.add_log_entry('D_diff', torch.mean(pred_d_fake.detach()) - torch.mean(pred_d_real.detach()))
|
||||||
if self.spsr_enabled and self.l_gan_grad_w > 0:
|
if self.spsr_enabled and self.cri_grad_gan > 0:
|
||||||
self.add_log_entry('l_d_real_grad', l_d_real_grad.detach().item())
|
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('l_d_fake_grad', l_d_fake_grad.detach().item())
|
||||||
self.add_log_entry('D_fake_grad', torch.mean(pred_d_fake_grad.detach()))
|
self.add_log_entry('D_fake_grad', torch.mean(pred_d_fake_grad.detach()))
|
||||||
|
|
Loading…
Reference in New Issue
Block a user