diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 7313d92339846f91c65826fe9b548baa597bf975..5f63ef37d62644eee2f6db496acb83bd4d15d5f7 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -619,10 +619,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()