0710-1537

diffusion.py

@@ -362,12 +362,13 @@ class DDPM21CM:
         # self.shape_loaded = dataset.images.shape
         # print("shape_loaded =", self.shape_loaded)
         self.dataloader = DataLoader(
-            dataset, 
+            dataset=dataset, 
             batch_size=self.config.batch_size, 
-            shuffle=True, 
+            shuffle=False, 
             num_workers=1,#len(os.sched_getaffinity(0)), 
             pin_memory=True,
             persistent_workers=True,
+            sampler=DistributedSampler(dataset),
             )
         # del dataset
         # self.accelerate(self.config)