Added the LayerPruning callback

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

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

78 stars 10 forks source link

Closed tarekziade closed 5 months ago

tarekziade commented 5 months ago

Adds a callback that can be used to prune the encoder and/or decoder layers