diff --git a/modules/deeplearning/cnn_l1b_l2_16.py b/modules/deeplearning/cnn_l1b_l2_16.py
index 0531a21beb099084b5f820b6d04667a7e46dcb03..3b46c1f249b52bd7af81f36f6164a18c27e2445c 100644
--- a/modules/deeplearning/cnn_l1b_l2_16.py
+++ b/modules/deeplearning/cnn_l1b_l2_16.py
@@ -216,12 +216,12 @@ class UNET:
         self.n_chans = 10
         if TRIPLET:
             self.n_chans *= 3
-        # self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
-        self.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
+        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
+        # self.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
 
         self.inputs.append(self.X_img)
-        # self.inputs.append(tf.keras.Input(shape=(None, None, self.n_chans)))
-        self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
+        self.inputs.append(tf.keras.Input(shape=(None, None, self.n_chans)))
+        # self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
 
         self.DISK_CACHE = False