From a0a295d24fa64bba1278b5201474575f4e9f155b Mon Sep 17 00:00:00 2001
From: zhuoyiyang <yangzhuo18@mails.tsinghua.edu.cn>
Date: Wed, 27 Oct 2021 11:39:16 +0800
Subject: [PATCH] fix out range bug
---
data_utils/configure_data.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/data_utils/configure_data.py b/data_utils/configure_data.py
index 4d44335..c34c435 100755
--- a/data_utils/configure_data.py
+++ b/data_utils/configure_data.py
@@ -298,6 +298,7 @@ class BlockedRandomSplitDataset(data.Dataset):
def __init__(self, ds, indices, block_size,**kwargs):
if type(indices) is not np.ndarray:
indices = np.array(indices)
+ indices = np.sort(indices)
self.block_size = block_size
self.wrapped_data = ds
self.wrapped_data_len = len(ds)
--
GitLab