0712-1535

diffusion.py

@@ -237,7 +237,7 @@ class TrainConfig:
     stride = (2,2) if dim == 2 else (2,2,1)
     num_image = 2000#32000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560
     batch_size = 10#2#50#20#2#100 # 10
-    n_epoch = 10#50#20#20#2#5#25 # 120
+    n_epoch = 5# 10#50#20#20#2#5#25 # 120
     HII_DIM = 28#64
     num_redshift = 2#128#64#512#256#256#64#512#128
     channel = 1
@@ -603,7 +603,7 @@ if __name__ == "__main__":
 
 def generate_samples(model, num_new_img, max_num_img_per_gpu, rank, world_size):
     samples = []
-    for _ in ranges(num_new_img // max_num_img_per_gpu):    
+    for _ in range(num_new_img // max_num_img_per_gpu):    
         sample = model.module.sample(filename, params=torch.tensor([4.4, 131.341]), num_new_img=max_num_img_per_gpu)
         samples.append(sample)
         # model.sample(filename, params=torch.tensor((5.6, 19.037)), num_new_img=max_num_img_per_gpu)