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?