From a464bba3b4749b8d8c685daa8fcd2101ca07cebd Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 3 Aug 2022 11:25:49 -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 0cb574dd..f39d303c 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -441,7 +441,7 @@ class ESPCN:
         conv = tf.keras.layers.LeakyReLU()(conv)
         print(conv.shape)
 
-        conv = tf.keras.layers.Conv2D(num_filters/2, kernel_size=3, strides=1, padding=padding, activation=None)(conv)
+        conv = tf.keras.layers.Conv2D(num_filters/2, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
         conv = tf.keras.layers.BatchNormalization()(conv)
         print(conv.shape)
 
-- 
GitLab