snapshot...

modules/GSOC/E2_ESRGAN/lib/utils.py

@@ -164,6 +164,12 @@ def pixel_loss(y_true, y_pred):
   return tf.reduce_mean(tf.reduce_mean(tf.abs(y_true - y_pred), axis=0))
 
 
+def pixel_loss_mse(y_true, y_pred):
+  y_true = tf.cast(y_true, tf.float32)
+  y_pred = tf.cast(y_pred, tf.float32)
+  return tf.reduce_mean(tf.reduce_mean(tf.square(y_true - y_pred), axis=0))
+
+
 def RelativisticAverageLoss(non_transformed_disc, type_="G"):
   """ Relativistic Average Loss based on RaGAN
       Args: