diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index cfff17537ffb155c214a4118280260ab3d62311d..dab247bbbe6387ab0c9c0e2913dcded235803e6d 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -231,6 +231,8 @@ class SRCNN:
         self.y_k = slice(1, LEN_Y + 3)
         self.x_128 = slice(4, LEN_X + 4)
         self.y_128 = slice(4, LEN_Y + 4)
+        self.LEN_X = LEN_X
+        self.LEN_Y = LEN_Y
 
         tf.debugging.set_log_device_placement(LOG_DEVICE_PLACEMENT)