snapshot...

modules/deeplearning/srcnn_l1b_l2.py

@@ -314,7 +314,7 @@ class SRCNN:
 
         trn_idxs = np.arange(len(train_data_files))
         np.random.shuffle(trn_idxs)
-        tst_idxs = np.arange(len(train_data_files))
+        tst_idxs = np.arange(len(test_data_files))
 
         self.get_train_dataset(trn_idxs)
         self.get_test_dataset(tst_idxs)