Is it impossible to pickle model with hook?

[malioboro (Rian Adam)]

Hi, I am facing a similar situation with these questions from different years:

• Torch.save Can't pickle local object
• Save model when the hooks are used
• Saving the model while using another class for the layers of that model

And all of them don’t have an answer yet. So, Is it impossible to pickle a PyTorch model that contains hook?

[marksaroufim]

Interesting, the easiest way to solve this is to torch.save(model.state_dict())

It might be possible to also pickle hooks if you wrap them in classes as a nice to have cc @albanD but regardless in the meantime state_dict() is the way to go

[albanD]

Serializing only the state_dict is the way around pickle’s limitations yes!

[malioboro (Rian Adam)]

I see… thank you for the clarification!