Xsmos
/

ml21cm

generate 21cm lightcones

denoising diffusion probabilistic model

Model card Files Files and versions

Metrics Training metrics Community

Xsmos commited on Jul 12, 2024

Commit

877d889

·

verified ·

1 Parent(s): 7cdcb6e

0712-1534

Files changed (1) hide show

diffusion.py +1 -1

diffusion.py CHANGED Viewed

@@ -652,7 +652,7 @@ if __name__ == "__main__":
         config.num_image = num_image
         # print("ddpm21cm = DDPM21CM(config)")
-        manager = np.Manager()
         return_dict = manager.dict()
         mp.spawn(sample, args=(world_size, config, num_new_img, max_num_img_per_gpu, return_dict), nprocs=world_size, join=True)

         config.num_image = num_image
         # print("ddpm21cm = DDPM21CM(config)")
+        manager = mp.Manager()
         return_dict = manager.dict()
         mp.spawn(sample, args=(world_size, config, num_new_img, max_num_img_per_gpu, return_dict), nprocs=world_size, join=True)