Issues with torch.utils.data.random_split

It worked. Thanks a lot :blush: