From 22becf3915fe1ef73dbbceeefb55d4e11d7eef14 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Jul 2019 15:23:39 -0400 Subject: [PATCH] added single node distdataparallel --- pytorch_lightning/models/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 9f473afd98..463a6cbd21 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -292,6 +292,7 @@ class Trainer(TrainerIO): def __getstate__(self): """ This is called before pickling. """ state = self.__dict__.copy() + del state['cluster'] return {} def __dp_train(self, gpu_nb, proc_rank):