I’m facing the same error with RandomResizedCrop. I’ll try the solution you said. Do we need to implement vision.transforms.ToPILImage() to use torchvision Transforms? Won’t numpy arrays be acceptable?