diff --git a/modules/GSOC/E2_ESRGAN/lib/utils.py b/modules/GSOC/E2_ESRGAN/lib/utils.py index e2c497b297e54a9af0df48aa4221d5fa102cb719..954910abae10c3bc831ca5bf274f776573570dea 100644 --- a/modules/GSOC/E2_ESRGAN/lib/utils.py +++ b/modules/GSOC/E2_ESRGAN/lib/utils.py @@ -205,6 +205,7 @@ def RelativisticAverageLoss(non_transformed_disc, type_="G"): fake_loss = tf.nn.sigmoid_cross_entropy_with_logits( labels=tf.ones_like(fake_logits), logits=fake_logits) return real_loss + fake_loss + if type_ == "G": loss = loss_G elif type_ == "D":