mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
removed logging
This commit is contained in:
@@ -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