If you are using multiple workers for the DataLoaders, try to use num_workers=0 and rerun the code.
If that’s working, try to run the code as a  script from your terminal instead of a notebook.
Let me know, if that doesn’t help.