From 4da4e52cd909ad94ee435f9bcb6edc5d202dc633 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 4 Mar 2023 11:59:28 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cnn_cld_frac_mod_res.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index 8cd62688..eb12f231 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -602,11 +602,11 @@ class SRCNN:
self.test_loss(t_loss)
self.test_accuracy(labels, pred)
- @tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
+ # @tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
def predict(self, inputs, labels):
labels = tf.squeeze(labels)
pred = self.model([inputs], training=False)
- t_loss = self.loss(labels, pred)
+ t_loss = self.loss(tf.squeeze(labels), pred)
self.test_labels.append(labels)
self.test_preds.append(pred.numpy())
--
GitLab