torchvision == 0.18.0
