Thanks, I actually did follow that. But I stopped because I was not quite sure what the collate function was doing from the utils.py file. I could not quite understand the reason it was used while loading the data. Could you please explain what it does and why it is needed?