From 5fbe00837e431615ef8bfcdd50a5665fe7115ed5 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 09:33:58 -0400 Subject: [PATCH] easy import for lightningModule --- pytorch_lightning/models/trainer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 942107d6..ef3a305b 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -324,12 +324,11 @@ class Trainer(TrainerIO): def __get_root_node_ip(self, proc_rank, nb_gpu_nodes, ip_file_dir): # on one node we use localhost - # if nb_gpu_nodes == 1: - # return '127.0.0.1' + if nb_gpu_nodes == 1: + return '127.0.0.1' # on multi-node, every node rank > 0 waits until rank 0 # saves the ip to disk - print(ip_file_dir) ip_file = os.path.join(ip_file_dir, '.ip_meta') if proc_rank == 0: # get the proc 0 IP