I shall take a look at creating the custom collate_fn
Regarding taking forever, I also meant that one process is blocking another.
If I don’t use num workers, my dataLoader with 8 threads take around 1.5 second to load the batch. But when I introduced num_workers=1(or any +ve number), even after 5-10 seconds nothing came up.