Torchvision latest version. There might be a mismatch.