snapshot...

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)