diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 4c3ba51d7ed8e98768800bb9782aece95f2711f9..990036ec429e04b3ea104ef2bbec34d7e31e5127 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -436,11 +436,11 @@ class ESPCN:
         conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=4, strides=2, padding=padding, activation=activation)(conv)
         print(conv.shape)
 
-        # conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
-        # print(conv.shape)
-        #
-        # conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
-        # print(conv.shape)
+        conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=1, strides=1, padding=padding, activation=activation)(conv)
+        print(conv.shape)
+
+        conv = tf.keras.layers.Conv2DTranspose(num_filters // 8, kernel_size=1, strides=1, padding=padding, activation=activation)(conv)
+        print(conv.shape)
 
         #self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability', activation=tf.nn.sigmoid)(conv)
         self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability')(conv)