diff --git a/data_utils/configure_data.py b/data_utils/configure_data.py index 4d44335db1d208289d3abbd01d59721db6acdc52..c34c435a73e52d92240caace86d995d36737fba0 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)