From b7baa961862ac022450800e335da121bd3c86b8c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 12 Jul 2019 16:29:44 -0400 Subject: [PATCH] fixed nccl init --- pytorch_lightning/models/trainer.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 68658c61..9c58c05b 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -339,10 +339,7 @@ class Trainer(TrainerIO): # set up server using proc 0's ip address # try to init for 20 times at max in case ports are taken # where to store ip_table - print('-'*100) - print('INIT CONN') self.__init_tcp_connection() - print('-'*100) # CHOOSE OPTIMIZER # filter out the weights that were done on gpu so we can load on good old cpus