Please see the here: https://pytorch.org/docs/stable/data.html#torch.utils.data.random_split and this post: https://discuss.pytorch.org/t/how-to-split-dataset-into-test-and-validation-sets/33987