diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 8594f731f2..eedf955baa 100644 --- 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