Torch distributed and dataset pickle error

replied in your other thread: Torch distributed and num_workers>0 pickle error - #6 by H-Huang