Solved by codearts in post #2 
              
              
                Never mind. I am blind. I thought it says 224x224 but it says 224x244 (which makes sense now). 
I used a resize transformation which fixes the issue in case of more images with unmatching dimensions. In my case I had just one image which was 224x244 so I fixed it manually and removed the resize tran…