diff --git a/pl_examples/loop_examples/kfold.py b/pl_examples/loop_examples/kfold.py index 811ad409c2e91..229673ec78df4 100644 --- a/pl_examples/loop_examples/kfold.py +++ b/pl_examples/loop_examples/kfold.py @@ -239,6 +239,9 @@ def __getattr__(self, key) -> Any: return getattr(self.fit_loop, key) return self.__dict__[key] + def __setstate__(self, state: Dict[str, Any]) -> None: + self.__dict__.update(state) + class LitImageClassifier(ImageClassifier): def __init__(self) -> None: