0715-1409

diffusion.py

@@ -240,8 +240,8 @@ class TrainConfig:
     num_image = 1000#32000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560
     batch_size = 1#2#50#20#2#100 # 10
     n_epoch = 3# 10#50#20#20#2#5#25 # 120
-    HII_DIM = 32
-    num_redshift = 128#128#64#512#256#256#64#512#128
+    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)