mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
removed logging
This commit is contained in:
1 parent
468bd141f4
commit
229d168c20
1 file changed
+1
-7
@@ -450,14 +450,8 @@ class Trainer(TrainerIO):
|
||||
# node rank using relative slurm id
|
||||
# otherwise default to node rank 0
|
||||
try:
|
||||
print('x'*100)
|
||||
node_id = os.environ['SLURM_NODEID']
|
||||
local_id = os.environ['SLURM_LOCALID']
|
||||
n_nodes = os.environ['SLURM_JOB_NUM_NODES']
|
||||
|
||||
print(f'NODEID: {node_id}, LOCALID: {local_id}, N_NODES: {n_nodes}')
|
||||
print('x'*100)
|
||||
self.node_rank = int(os.environ['SLURM_NODEID'])
|
||||
self.node_rank = int(node_id)
|
||||
except Exception as e:
|
||||
self.node_rank = 0
|
||||
|
||||
|
||||
Reference in new issue
Block a user