Hi Preetham,
thanks, the detection models work with any sizes. I actually found a solution for the above, it was indeed the dataloader and using a custom_collate function worked.
Though forks (i.e. parallel execution) doesn’t with my custom collate_fn, which is a bit of a shame.