From 5d908798a555a71f677246cb185af4a11f2e9ca2 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 19 Oct 2022 10:35:35 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/srcnn_l1b_l2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index fd525c97..e845ebf6 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -645,8 +645,8 @@ class SRCNN: return pred def run(self, directory, ckpt_dir=None, num_data_samples=50000): - train_data_files = glob.glob(directory+'data_train*.npy') - valid_data_files = glob.glob(directory+'data_valid*.npy') + train_data_files = glob.glob(directory+'data_train_*.npy') + valid_data_files = glob.glob(directory+'data_valid_*.npy') self.setup_pipeline(train_data_files, valid_data_files, num_data_samples) self.build_model() -- GitLab