Based on the error message it seems the collate_fn fails to index the returned batch samples.
Are you able to iterate the DataLoader for an entire epoch without any model training? If not, could you use num_workers=0 and retry it?