Made printing for DDP init debug only

mala-project · May 2, 2024 · d9c7a73 · d9c7a73
1 parent 36e626c
commit d9c7a73
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mala/common/parameters.py b/mala/common/parameters.py
@@ -1300,7 +1300,8 @@ def use_ddp(self):
     @use_ddp.setter
     def use_ddp(self, value):
         if value:
-            print("initializing torch.distributed.")
+            if self.verbosity > 1:
+                print("Initializing torch.distributed.")
             # JOSHR:
             # We start up torch distributed here. As is fairly standard
             # convention, we get the rank and world size arguments via

diff --git a/mala/datahandling/data_handler.py b/mala/datahandling/data_handler.py
@@ -640,7 +640,7 @@ def __build_datasets(self):
                     self.descriptor_calculator,
                     self.target_calculator,
                     self.use_ddp,
-                    self.parameters._configuration["device"]
+                    self.parameters._configuration["device"],
                 )
             )
             self.validation_data_sets.append(
@@ -652,7 +652,7 @@ def __build_datasets(self):
                     self.descriptor_calculator,
                     self.target_calculator,
                     self.use_ddp,
-                    self.parameters._configuration["device"]
+                    self.parameters._configuration["device"],
                 )
             )