Solved by ptrblck in post #4 
              
              
                Could you specify the size in Resize as a tuple? 
transforms.Resize((224, 224))

as a single value will work differently, if your images are not quadratic.