This example throws this error "ImportError: cannot import name '_get_image_size' from 'torchvision.transforms.functional'"
Related to this
Fixed by this PR.