diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 1ff295c93b373dd0b8ba44ebd67fbda960b1c917..6f0e16480c449cbe128b908f22a70645ffe60d0e 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -91,6 +91,11 @@ if KERNEL_SIZE == 3:
     slc_y = slice(0, int(Y_LEN/4) + 2)
     x_64 = slice(4, X_LEN + 4)
     y_64 = slice(4, Y_LEN + 4)
+elif KERNEL_SIZE == 1:
+    slc_x = slice(4, X_LEN + 4)
+    slc_y = slice(4, Y_LEN + 4)
+    x_64 = slice(4, X_LEN + 4)
+    y_64 = slice(4, Y_LEN + 4)
 # ----------------------------------------
 
 
@@ -515,7 +520,7 @@ class SRCNN:
 
         conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_6', kernel_size=KERNEL_SIZE, scale=scale)
 
-        conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
+        conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
 
         # conv = conv + conv_b
         conv = conv_b