removed logging

This commit is contained in:
William Falcon
2019-07-20 09:15:09 -04:00
parent 468bd141f4
commit 229d168c20
+1 -7
View File
@@ -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