diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index a54e5abe5389f7ea8430f51432866cc5f89c7c4d..2c8305d1f2e683f6004a33e3a56ae95e44a1f0ca 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -627,10 +627,7 @@ class SRCNN: train_data_files = glob.glob(directory+'data_train*.npy') valid_data_files = glob.glob(directory+'data_valid*.npy') - train_data_files.sort() - valid_data_files.sort() - - self.setup_pipeline(train_data_files, valid_data_files, 100000) + self.setup_pipeline(train_data_files, valid_data_files, 50000) self.build_model() self.build_training() self.build_evaluation()