Skip to content

Commit 1d2e783

Browse files
committed
fix: don't print devices twice
1 parent 39fd5e9 commit 1d2e783

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

‎dmlcloud/core/callbacks.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -460,10 +460,6 @@ def pre_run(self, pipe):
460460

461461
diagnostics = general_diagnostics()
462462

463-
diagnostics += '\n* DEVICES:\n'
464-
devices = all_gather_object(str(pipe.device))
465-
diagnostics += '\n'.join(f' - [Rank {i}] {device}' for i, device in enumerate(devices))
466-
467463
diagnostics += '\n* CONFIG:\n'
468464
diagnostics += '\n'.join(f' {line}' for line in OmegaConf.to_yaml(pipe.config, resolve=True).splitlines())
469465

0 commit comments

Comments
 (0)