If I understand correctly, you can find info about image sizes in the webpage of each dataset. If you are using your own custom dataset, you can check it in __getitem__ method by using CustomDataset. Here is post for such a situation: Custom Dataset with some preprocessing
Also, dataloading tutorial may help.
But a simple solution would be adding transforms.Resize(224) in the previous post right before ToTensor().