Adjusted output for parallel

mala-project · Apr 26, 2024 · 04b0050 · 04b0050
1 parent ffa3082
commit 04b0050
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mala/network/trainer.py b/mala/network/trainer.py
@@ -45,7 +45,7 @@ def __init__(self, params, network, data, optimizer_dict=None):
         super(Trainer, self).__init__(params, network, data)
 
         if self.parameters_full.use_ddp:
-            print("wrapping model in ddp..")
+            printout("DDP activated, wrapping model in DDP.", min_verbosity=1)
             # JOSHR: using streams here to maintain compatibility with
             # graph capture
             s = torch.cuda.Stream()