From d9c7a73562c6798513ff06bf3c6c5db3e28f468b Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Thu, 2 May 2024 08:40:13 +0200 Subject: [PATCH] Made printing for DDP init debug only --- mala/common/parameters.py | 3 ++- mala/datahandling/data_handler.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) 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"], ) )