I\'m trying to create a custom dataset, but it fails in transform.
bellow is my code:
transform = transforms.Compose([transforms.ToPILImage(),