@kirk86 Can you share your key code that loading different size input images?
I read the source code of dataloader, finding that torch.stack()  is used. This function expects all elements in the batch sequence the exactly same size. How do you handle that?
@smth
If I set batch size to greater than 1, how can I use a dataloader of different size inputs?
Thanks.