From a946483f1c40016250789c2e2895e2388addcbe6 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Thu, 14 May 2020 20:45:06 -0600
Subject: [PATCH] Fix discriminator noise floor
---
codes/models/SRGAN_model.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py
index d1283232..3183d4d9 100644
--- a/codes/models/SRGAN_model.py
+++ b/codes/models/SRGAN_model.py
@@ -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 = []