diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index f372878fb1729d6aba08a992231753985ac22724..5c6f30c91afdd7da113003a66956e5c57a035f20 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -210,9 +210,9 @@ class ESPCN:
 
         self.n_chans = 1
 
-        # self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
+        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
         # self.X_img = tf.keras.Input(shape=(36, 36, self.n_chans))
-        self.X_img = tf.keras.Input(shape=(32, 32, self.n_chans))
+        # self.X_img = tf.keras.Input(shape=(32, 32, self.n_chans))
 
         self.inputs.append(self.X_img)