Hi,
Thank you for the reply and my sincere apologies for the late response. That’s kinda why I want to create a custom dataloader because I am not working with a generic dataset and getting the dataset into a generic one will take a lot of time.
I think I was looking for this only, collate_fn. Probably this is why I am getting this error.