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?