Skip to content
Snippets Groups Projects
Commit 62991ccc authored by tomrink's avatar tomrink
Browse files

minor

parent a076f9da
Branches
No related tags found
No related merge requests found
......@@ -194,13 +194,13 @@ class SRCNN:
def get_in_mem_data_batch(self, idxs, is_training):
if is_training:
label_files = self.train_data_files
files = self.train_data_files
else:
label_files = self.test_data_files
files = self.test_data_files
label_s = []
for k in idxs:
f = label_files[k]
f = files[k]
nda = np.load(f)
label_s.append(nda)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment