diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index bf690cf89707aefb52e5349f2584fd0535971bc4..df363c4b04977d889f8d573fc0ba6dc974e9b4c5 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -612,7 +612,8 @@ class SRCNN:
     # decorator commented out because pred.numpy(): pred not evaluated yet.
     def predict(self, inputs, labels):
         pred = self.model([inputs], training=False)
-        t_loss = self.loss(tf.squeeze(labels), pred)
+        # t_loss = self.loss(tf.squeeze(labels), pred)
+        t_loss = self.loss(labels, pred)
 
         self.test_labels.append(labels)
         self.test_preds.append(pred.numpy())