From 229d168c2016ba506550405d0be23db117c12084 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 20 Jul 2019 09:15:09 -0400 Subject: [PATCH] removed logging --- pytorch_lightning/models/trainer.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 5b518283..f153d6a9 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -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