torchprune
torchprune copied to clipboard
export to onnx
Hi Thanks for sharing this awesome code!
I noticed some customize pytorch modules is used in the trained pytorch models. I'm wondering if one could export the fine-tuned model to onnx after, for example, applying ALDS methods? Or is there any plan to support this?
It would be great if the trained model can be exported to onnx for deployment.
Hi, Thanks for raising this issue. That's a great idea!
It should be really easy to implement since all the checkpoints are handled by the trainer here.
I will implement it once I have some time to take care of it - otherwise, I would be happy to take PRs as well.