diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 7bf6cd23389602d99b4d18611821348cf74b47f4..4bcbdc5e71fff60e7bdeb89be9c8df3d2cd6d6d1 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -682,8 +682,8 @@ class SRCNN: 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 = train_data_files[::2] - valid_data_files = valid_data_files[::2] + # train_data_files = train_data_files[::2] + # valid_data_files = valid_data_files[::2] self.setup_pipeline(train_data_files, valid_data_files, num_data_samples) self.build_model()