Your model implies a specific input size. So, if your training set images are a different size, you’ll have to preprocess them to the correct size (somehow) before you can feed them into your model. See https://pytorch.org/tutorials/beginner/data_loading_tutorial.html