Daniel_Dsouza:

Shouldn’t this be the default ? Actually maybe I get what you’re sayin. Is it like "Collate is where I’d get it ready for the model directly ?


I’m not sure I understand the question correctly.
The collate function gets the samples from multiple calls into Dataset.__getitem__ and creates a batch, which you will get in the DataLoader loop.
If you are fine with the current shape, then there is no need to write a custom collate function.
However, if you would like to change something,  have a look at this doc about the collate_fn.