Running inference with varying input size

You have to define a custom transform on your dataset where the different shape inputs are handled. The simplest solution would be padding. You can find some more details in this thread: Can data loader work with different input shape