I see the code in utils.py idx_train = range(140) idx_val = range(200, 500) idx_test = range(500, 1500) I cannot understand it, because the ratio of they may be 6:2:2 if the dataset is smaller. Anyone can help me?