diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index 00291598d6bd04a8f5150c822c28792cf47a8964..87ee61a204082929fff73f8bed0b103c32c3c3ef 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -801,10 +801,10 @@ class SRCNN:
         return pred
 
     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_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')
         self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, num_data_samples)
 
         # train_data_files = glob.glob(directory+'data_train_*.npy')