snapshot..

modules/deeplearning/srcnn_l1b_l2.py

@@ -694,7 +694,7 @@ class SRCNN:
         self.do_training(ckpt_dir=ckpt_dir)
 
     def run_restore(self, directory, ckpt_dir):
-        valid_data_files = glob.glob(directory + 'data_*.npy')
+        valid_data_files = glob.glob(directory + 'data_valid*.npy')
         self.num_data_samples = 1000
         self.setup_test_pipeline(valid_data_files)
         self.build_model()