04145638

diffusion.py

@@ -505,7 +505,7 @@ class DDPM21CM:
 
         # print("!!!!!!!!!!!!!!!!, before prepare, self.dataloader.sampler =", self.dataloader.sampler)
         print_start = time()
-        print(f"model: {self.nn_model.device}, cuda:{torch.cuda.current_device()}/{self.config.global_rank}") 
+        print(f"model: {self.nn_model.device}")
         #print(f"optimizer: {self.optimizer.state_dict()}")
         print(f"dataloader: {next(iter(self.dataloader))[0].device}")
         print(f"lr_scheduler: {self.lr_scheduler.optimizer is self.optimizer}")