JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡
https://julesbelveze.github.io/bert-squeeze/
79 stars 10 forks source link

feat: add CustomLabse #15

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

This PR aims at implementing a custom version LaBSE using the CustomBertEncoder to be able to perform layer dropping with LaBSE