JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

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

Fix `BertCustomEncoder`

JulesBelveze opened this issue 3 years ago · comments

Jules Belveze commented 3 years ago

Output of BertCustomEncoder seems to differ from the regular BertEncoder

Jules Belveze commented 3 years ago

Fixed in #15