torchvision == 0.18.1
