snapshot...

modules/GSOC/E2_ESRGAN/lib/utils.py

@@ -216,14 +216,14 @@ def RelativisticAverageLoss(non_transformed_disc, type_="G"):
 
 
 # Standard GAN loss functions ---------------------------------------
-def gen_loss(non_transformed_disc, fake_image):
+def std_loss_G(non_transformed_disc, fake_image):
     fake_logits = non_transformed_disc(fake_image)
     fake_loss = tf.nn.sigmoid_cross_entropy_with_logits(
         labels=tf.ones_like(fake_logits), logits=fake_logits)
     return fake_loss
 
 
-def disc_loss(non_transformed_disc, real_image, fake_image):
+def std_loss_D(non_transformed_disc, real_image, fake_image):
     real_logits = non_transformed_disc(real_image)
     fake_logits = non_transformed_disc(fake_image)