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