From efbdf40fa1de40ef4a627bb05b2f32f1562f7cc0 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 18 Apr 2022 12:00:51 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_fcn.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 552d533e..e00c19e7 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -736,6 +736,7 @@ class IcingIntensityFCN:
inputs = [mini_batch[0], mini_batch[1]]
labels = mini_batch[2]
pred = self.model(inputs, training=False)
+ pred = tf.reshape(pred, (pred.shape[0], NumLogits))
t_loss = self.loss(labels, pred)
self.test_labels.append(labels)
--
GitLab