From 98f9bdeaf9ba1f04beb4418aa92357cda36dc4e9 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 1 Aug 2022 13:39:53 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 4086e006..2f02c9a2 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -451,7 +451,7 @@ class ESPCN:
activation = tf.nn.softmax # For multi-class
# Called logits, but these are actually probabilities, see activation
- self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability', activation=activation)(conv)
+ self.logits = tf.keras.layers.Activation(activation=activation)(conv)
print(self.logits.shape)
--
GitLab