From c54784ae9ed0b3cd4cecbdc1e1060a0715e90f87 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 26 Jul 2020 22:25:59 -0600 Subject: [PATCH] Fix feature disc log item error --- codes/models/SRGAN_model.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index c651dbad..677160d1 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -540,16 +540,16 @@ class SRGANModel(BaseModel): if self.l_gan_w > 0: self.add_log_entry('l_g_gan', l_g_gan_log.item()) self.add_log_entry('l_g_total', l_g_total_log.item()) + if self.opt['train']['gan_type'] == 'pixgan_fea': + self.add_log_entry('l_d_fea_fake', l_d_fea_fake.item() * self.mega_batch_factor) + self.add_log_entry('l_d_fea_real', l_d_fea_real.item() * self.mega_batch_factor) + self.add_log_entry('l_d_fake_total', l_d_fake.item() * self.mega_batch_factor) + self.add_log_entry('l_d_real_total', l_d_real.item() * self.mega_batch_factor) if self.l_gan_w > 0 and step > self.G_warmup: self.add_log_entry('l_d_real', l_d_real_log.item()) self.add_log_entry('l_d_fake', l_d_fake_log.item()) self.add_log_entry('D_fake', torch.mean(pred_d_fake.detach())) self.add_log_entry('D_diff', torch.mean(pred_d_fake) - torch.mean(pred_d_real)) - if self.opt['train']['gan_type'] == 'pixgan_fea': - self.add_log_entry('l_d_fea_fake', l_d_fea_fake.item() * self.mega_batch_factor) - self.add_log_entry('l_d_fea_real', l_d_fea_real.item() * self.mega_batch_factor) - self.add_log_entry('l_d_fake_total', l_d_fake.item() * self.mega_batch_factor) - self.add_log_entry('l_d_real_total', l_d_real.item() * self.mega_batch_factor) # Log learning rates. for i, pg in enumerate(self.optimizer_G.param_groups):