diff --git a/mala/common/parameters.py b/mala/common/parameters.py index 6a8baec76..3627bd40f 100644 --- 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 index 266664e59..7b8fc2a43 100644 --- 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"], ) )