From 30f0fdc8612ac70d14fb3cb6a6908873ff271643 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 7 Mar 2022 11:23:40 -0600
Subject: [PATCH] minor...
---
modules/deeplearning/icing_fcn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 9002f290..5f0ba197 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -627,7 +627,7 @@ class IcingIntensityFCN:
activation = tf.nn.softmax # For multi-class
# Called logits, but these are actually probabilities, see activation
- logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, activation=activation)(conv)
+ logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability', activation=activation)(conv)
print(logits.shape)
--
GitLab