Ah. So, with random_split, I wouldn’t be able, for instance, do a stratified split based on output labels in case of imbalanced data? But I would able to do the same with SubsetRandomSampler/ WeightedRandomSampler?