guided-diffusion
guided-diffusion copied to clipboard
stuck in dist_util.setup_dist() when trying multiGPUs training.
I use python -m torch.distributed.launch --nproc_per_node=4 ... to train in a multiGPU way, but it is stuck in "dist_util.setup_dist()" which is the first sentence of main. Any body know the reason?