Xsmos
/

ml21cm

generate 21cm lightcones

denoising diffusion probabilistic model

Model card Files Files and versions

Metrics Training metrics Community

Xsmos commited on Jul 25, 2024

Commit

4e925ec

·

verified ·

1 Parent(s): a70d97d

0725-1814

Files changed (1) hide show

diffusion.py +2 -2

diffusion.py CHANGED Viewed

@@ -28,9 +28,9 @@
 # %%
 import logging
-#logging.getLogger("torch").setLevel(logging.ERROR)
 import warnings
-warnings.filterwarnings("ignore", message=r"^Detected kernel version")
 from dataclasses import dataclass
 import h5py

 # %%
 import logging
+logging.getLogger("torch").setLevel(logging.ERROR)
 import warnings
+#warnings.filterwarnings("ignore", message=r"^Detected kernel version")
 from dataclasses import dataclass
 import h5py