added single node distdataparallel

2019-07-03 15:25:56 -04:00 · 2019-07-03 15:25:56 -04:00 · 09ed6904c2
parent 970d1609e1
commit 09ed6904c2
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -292,7 +292,7 @@ class Trainer(TrainerIO):
    def __getstate__(self):
        """ This is called before pickling. """
        state = self.__dict__.copy()
-        del state['cluster']
+        # del state['cluster']
        del state['experiment']
        return state