Thank you so much for helping me.  I have learned what was going wrong. Just one question, the model expects a Tensor of shape [batch size,3,224,224]. But I’m passing a Tensor of shape [batch_size,3,256,128] . How can I make model accept tensor of different dim ?