torchvision == 0.17.1
