fix: don't print devices twice

sehoffmann · Jan 6, 2025 · 1d2e783 · 1d2e783
1 parent 39fd5e9
commit 1d2e783
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/dmlcloud/core/callbacks.py b/dmlcloud/core/callbacks.py
@@ -460,10 +460,6 @@ def pre_run(self, pipe):
 
         diagnostics = general_diagnostics()
 
-        diagnostics += '\n* DEVICES:\n'
-        devices = all_gather_object(str(pipe.device))
-        diagnostics += '\n'.join(f'    - [Rank {i}] {device}' for i, device in enumerate(devices))
-
         diagnostics += '\n* CONFIG:\n'
         diagnostics += '\n'.join(f'    {line}' for line in OmegaConf.to_yaml(pipe.config, resolve=True).splitlines())