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

Turn down feadisc intensity

Browse Source 
Honestly - this feature is probably going to be removed soon, so backwards
compatibility is not a huge deal anymore.
This commit is contained in:
James Betker 2020-07-27 15:28:55 -06:00
parent ebb199e884
commit 85ee64b8d9
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

@ -389,7 +389,7 @@ class SRGANModel(BaseModel):
l_d_fea_fake = 0 l_d_fea_fake = 0
if self.opt['train']['gan_type'] == 'pixgan_fea': if self.opt['train']['gan_type'] == 'pixgan_fea':
# Compute a feature loss which is added to the GAN loss computed later to guide the discriminator better. # Compute a feature loss which is added to the GAN loss computed later to guide the discriminator better.
disc_fea_scale = .5 disc_fea_scale = .1
_, fea_real = self.netD(var_ref, output_feature_vector=True) _, fea_real = self.netD(var_ref, output_feature_vector=True)
actual_fea = self.netF(var_ref) actual_fea = self.netF(var_ref)
l_d_fea_real = self.cri_fea(fea_real, actual_fea) * disc_fea_scale / self.mega_batch_factor l_d_fea_real = self.cri_fea(fea_real, actual_fea) * disc_fea_scale / self.mega_batch_factor