Where to actually specify the training size?

[davodogster]

Hi I'm a bit confused which size is used for training when I set this config and then trained the model:

It is confusing as there are lots of shape params.

And when I load the model for inference, which size will it reshape the image to?

Thanks, Sam (The model trained well)