I had earlier trained on 300 by 300 images. and yes I got it from pytorch model zoo. But now it only takes in 224x224 images… It is unable to take 300x300 images, or any other size . Only 224x224 images.