From cff0500a638860f22c0e3892cd3acbcd615971e8 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 82a24e9e04..87478726a9 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -343,10 +343,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