Khadijeh_Arabi:

I tried to use pytorch dataloader but I got this error:


Based on your code snippet you are not using a DataLoader, but are creating a custom generator, so I’m unsure where the error is coming from.
Could you post a minimal, executable code snippet to reproduce the issue?