This an issue from pyotch so you should ask it on their forum. From what I understand it has to do with shared memory while multiprocessing and setting num_workers to 0 will fix the problem (but make your training slower).
2 Likes