Issues with torch.utils.data.random_split

vision
12

Your code snippet works for me (PyTorch 1.4.0.dev20191109, torchvision 0.5.0a0+28003e9) and yields:

print(len(train_set))
> 50000
print(len(val_set))
> 10000

Which versions are you using?

2 Likes