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

Log total gen loss

Browse Source
This commit is contained in:
James Betker 2020-04-22 14:02:10 -06:00
parent 79aff886b5
commit ea5f432f5a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/SRGAN_model.py
View File

@ -224,10 +224,9 @@ class SRGANModel(BaseModel):
if self.cri_fea: if self.cri_fea:
self.log_dict['l_g_fea'] = l_g_fea.item() self.log_dict['l_g_fea'] = l_g_fea.item()
self.log_dict['l_g_gan'] = l_g_gan.item() self.log_dict['l_g_gan'] = l_g_gan.item()
self.log_dict['l_g_total'] = l_g_total.item()
self.log_dict['l_d_real'] = l_d_real.item() self.log_dict['l_d_real'] = l_d_real.item()
self.log_dict['l_d_fake'] = l_d_fake.item() self.log_dict['l_d_fake'] = l_d_fake.item()
self.log_dict['D_real'] = torch.mean(pred_d_real.detach())
self.log_dict['D_fake'] = torch.mean(pred_d_fake.detach()) self.log_dict['D_fake'] = torch.mean(pred_d_fake.detach())
def test(self): def test(self):