snapshot...

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())