Add an onnx exporter - Githubissues

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

https://julesbelveze.github.io/bert-squeeze/

78 stars 10 forks source link

Open tarekziade opened 4 months ago

tarekziade commented 4 months ago

Let's add a callback to write the model in the onnx format

JulesBelveze commented 4 months ago

Sounds really good! 🤓