added single node distdataparallel

2019-07-03 15:23:39 -04:00 · 2019-07-03 15:23:39 -04:00 · 22becf3915
parent 5797f812ad
commit 22becf3915
1 changed files with 1 additions and 0 deletions
--- 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):