Lazily loading inputs of different sizes without padding

[ptrblck]

@MHSA

However, I don’t have access to this information inside the Dataset.

That’s correct in the default use case. However, you could use a BatchSampler, which would allow you to get all indices of the current batch in the __getitem__ and apply your resizing logic as described e.g. here.