Could you use num_workers=0 and rerun the script?
If that’s working, could you iterate the Dataset directly without wrapping it into a DataLoader?
PS: It’s better to post code directly by wrapping it into three backticks ```, as it makes debugging easier.