diff --git a/pytorch_lightning/utilities/parsing.py b/pytorch_lightning/utilities/parsing.py index 8c43fd21d1..78131e2eec 100644 --- a/pytorch_lightning/utilities/parsing.py +++ b/pytorch_lightning/utilities/parsing.py @@ -87,7 +87,7 @@ def is_picklable(obj: object) -> bool: try: pickle.dumps(obj) return True - except (pickle.PickleError, AttributeError): + except (pickle.PickleError, AttributeError, RuntimeError): return False