Hi and thanks for replying,



 tymokvo:

Try using the torchvision.transforms.Resize with a size of (1024, 1024) on your input and see if that helps.


I thought using transforms.CenterCrop(image_size)  should do the trick;  but will try what you propose anyhow