I changed the dimension many way. But not a single change work for me. Here is my code
#transform for inceptionV3 train_transform = transforms.Compose([