Could you try to isolate which batch creates this issue or is it thrown in the first step of your DataLoader iteration?
You could iterate your Dataset and check, which index is raising this error.
Once you’ve isolated it, you could check what’s going wrong in the __getitem__ (or where this exception is raised).