Model checkpointing.

[kuntiik]

Hello there,
Has anyone encountered an error, when you save a .ckpt file and when you try to load it (for example using torch.load()) you get: 'Model' object has no attribute 'param_groups_fn'. This happens when torch tries to unpickle the file.
I've been working with PL for a while and so far had no problem with checkpointing. Can somebody please point me to documentation, on how to solve this or provide any insights into this error?
Thank you

[rohitgr7]

hey @kuntiik

haven't encountered this error before. Can you reproduce it with BoringModel. If it comes up with distributed training, you can still share a minimal script.