I’ve seen many people having this issue and the same for me too. I’m loading large images 2562563 size. Batch size is 32, with 1 worker. However, the loader stops loading data after one epoch, why does that happen?
Could you post a code snippet with a reproducible error?
Do you have this issue using your specific dataset?
Also, did you see this problem mentioned in this forum? If so, was there a solution posted?
What PyTorch version are you using?
If you are using OpenCV, check this issue: https://github.com/pytorch/pytorch/issues/1838