diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index 730207e37fc3aff40a41ffd5fea2ad010e65c006..24384512a3230d2b32929815b19bf99da6c8bd1d 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -658,8 +658,12 @@ class SRCNN:
     def run(self, directory, ckpt_dir=None, num_data_samples=50000):
         train_data_files = glob.glob(directory+'train*mres*.npy')
         valid_data_files = glob.glob(directory+'valid*mres*.npy')
-        train_label_files = glob.glob(directory+'train*ires*.npy')
-        valid_label_files = glob.glob(directory+'valid*ires*.npy')
+
+        # train_label_files = glob.glob(directory+'train*ires*.npy')
+        # valid_label_files = glob.glob(directory+'valid*ires*.npy')
+
+        train_label_files = [f.replace('mres', 'ires'), f in train_data_files]
+        valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
         self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, num_data_samples)
 
         self.build_model()
@@ -671,7 +675,8 @@ class SRCNN:
         self.num_data_samples = 1000
 
         valid_data_files = glob.glob(directory + 'valid*mres*.npy')
-        valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+        # valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+        valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
         self.setup_test_pipeline(valid_data_files, valid_label_files)
 
         self.build_model()