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
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