0712-1604

diffusion.py

@@ -639,7 +639,7 @@ if __name__ == "__main__":
     print(" sampling ".center(100,'-'))
     world_size = torch.cuda.device_count()
     # num_image_list = [1600,3200,6400,12800,25600]
-    num_image_list = [1000]
+    num_image_list = [100]
     num_new_img = 12
     max_num_img_per_gpu = 2
 
@@ -650,7 +650,7 @@ if __name__ == "__main__":
 
     for num_image in num_image_list:
         config.num_image = num_image
-        config.resume = f"./outputs/model_state-N{num_image}-epoch9-device0"
+        config.resume = f"./outputs/model_state-N{num_image}-epoch4-device0"
 
         # print("ddpm21cm = DDPM21CM(config)")
         manager = mp.Manager()