diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index d189b523ffc337a6856032abfa7aafefaecfd5e3..4c3ba51d7ed8e98768800bb9782aece95f2711f9 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -433,11 +433,7 @@ class ESPCN:
         # conv = tf.keras.layers.Conv2D(num_filters // 2, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
         # print(conv.shape)
 
-        conv = tf.keras.layers.Conv2D(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=2, padding=padding, activation=activation)(conv)
-        conv = tf.keras.layers.Conv2DTranspose(1, kernel_size=3, strides=2, padding=padding, activation=activation)(conv)
+        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)