Hello,
Looking at the docs 34 for transforms.Resize, you need to specify a sequence (h, w) if you want to reshape the image in both dimensions. If you only specify one number (like you did), it will resize the smaller edge and keep the aspect ratio for the other one.
Good luck!