From d540d476a04b63cb2de881e98a60351a685a045a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 09:29:02 -0400 Subject: [PATCH] easy import for lightningModule --- pytorch_lightning/models/trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 89713111..de111568 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -327,10 +327,9 @@ class Trainer(TrainerIO): # if nb_gpu_nodes == 1: # return '127.0.0.1' - pdb.set_trace() - # 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: os.makedirs(ip_file, exist_ok=True)