Xsmos commited on
Commit
f8f5f14
·
verified ·
1 Parent(s): ba47dea
Files changed (2) hide show
  1. diffusion.py +1 -1
  2. quantify_results.ipynb +0 -0
diffusion.py CHANGED
@@ -241,7 +241,7 @@ class TrainConfig:
241
  stride = (2,2) if dim == 2 else (2,2,2)
242
  num_image = 1000#2000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560
243
  batch_size = 50#1#2#50#20#2#100 # 10
244
- n_epoch = 60#5#4# 10#50#20#20#2#5#25 # 120
245
  HII_DIM = 64
246
  num_redshift = 64#512#128#64#512#256#256#64#512#128
247
  channel = 1
 
241
  stride = (2,2) if dim == 2 else (2,2,2)
242
  num_image = 1000#2000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560
243
  batch_size = 50#1#2#50#20#2#100 # 10
244
+ n_epoch = 120#5#4# 10#50#20#20#2#5#25 # 120
245
  HII_DIM = 64
246
  num_redshift = 64#512#128#64#512#256#256#64#512#128
247
  channel = 1
quantify_results.ipynb CHANGED
The diff for this file is too large to render. See raw diff