Xsmos
/

ml21cm

generate 21cm lightcones

denoising diffusion probabilistic model

Model card Files Files and versions

Metrics Training metrics Community

Xsmos commited on Jul 15, 2024

Commit

de75dea

·

verified ·

1 Parent(s): 45e791b

0715-1410

Files changed (1) hide show

diffusion.py +1 -1

diffusion.py CHANGED Viewed

@@ -241,7 +241,7 @@ class TrainConfig:
     batch_size = 1#2#50#20#2#100 # 10
     n_epoch = 3# 10#50#20#20#2#5#25 # 120
     HII_DIM = 64
-    num_redshift = 256#128#64#512#256#256#64#512#128
     channel = 1
     img_shape = (channel, HII_DIM, num_redshift) if dim == 2 else (channel, HII_DIM, HII_DIM, num_redshift)

     batch_size = 1#2#50#20#2#100 # 10
     n_epoch = 3# 10#50#20#20#2#5#25 # 120
     HII_DIM = 64
+    num_redshift = 512#128#64#512#256#256#64#512#128
     channel = 1
     img_shape = (channel, HII_DIM, num_redshift) if dim == 2 else (channel, HII_DIM, HII_DIM, num_redshift)