Solved by tom in post #2 
              
              
                I would start with the TorchVision documentation of Resize. It explains 

If size is a sequence like (h, w), output size will be matched to this.
If size is an int, smaller edge of the image will be matched to this number. i.e, if height  width, then image will be rescaled to (size * height / width,…