Issues with torch.utils.data.random_split

I’ve created a script which is given here: How to split dataset into test and validation sets