1 parent 39fd5e9 commit 1d2e783Copy full SHA for 1d2e783
1 file changed
dmlcloud/core/callbacks.py
@@ -460,10 +460,6 @@ def pre_run(self, pipe):
460
461
diagnostics = general_diagnostics()
462
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
-
467
diagnostics += '\n* CONFIG:\n'
468
diagnostics += '\n'.join(f' {line}' for line in OmegaConf.to_yaml(pipe.config, resolve=True).splitlines())
469
0 commit comments