added slurm no process warning

2026-09-11 12:31:23 +08:00 · 2019-07-18 17:05:09 -04:00
parent c02b6c4c88
commit 4e67983f23
1 changed files with 3 additions and 0 deletions
@@ -449,6 +449,9 @@ class Trainer(TrainerIO):
        except KeyError as e:
            node_rank = 0

+        print('-'*100)
+        print(gpu_nb)
+        print('-'*100)
        # recover original exp before went into process
        # init in write mode only on proc 0
        self.experiment.debug = self.proc_rank > 0