diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index 5ef886ed770e40b3a37d82ee1b515587f99bf7b9..39ab18a2e51999d3f0ce23d45a0a25a5235c9982 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -607,7 +607,7 @@ 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, axis=[3]), pred)
         t_loss = self.loss(labels, pred)
 
         self.test_labels.append(labels)