Fix `BertCustomEncoder`

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

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

78 stars 10 forks source link

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

Output of BertCustomEncoder seems to differ from the regular BertEncoder

JulesBelveze commented 2 years ago

Fixed in #15