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

Fix discriminator noise floor

Browse Source
This commit is contained in:
James Betker 2020-05-14 20:45:06 -06:00
parent c8ab89d243
commit a946483f1c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/SRGAN_model.py
View File

@ -198,10 +198,10 @@ class SRGANModel(BaseModel):
p.requires_grad = False
# Calculate a standard deviation for the gaussian noise to be applied to the discriminator, termed noise-theta.
if step >= self.D_noise_final:
if self.D_noise_final == 0:
noise_theta = 0
else:
noise_theta = (self.D_noise_theta - self.D_noise_theta_floor) * (self.D_noise_final - step) / self.D_noise_final + self.D_noise_theta_floor
noise_theta = (self.D_noise_theta - self.D_noise_theta_floor) * (self.D_noise_final - min(step, self.D_noise_final)) / self.D_noise_final + self.D_noise_theta_floor
self.fake_GenOut = []
var_ref_skips = []