JulesBelveze / bert-squeeze

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

Refacto: lr models #1

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

This PR aims at refactoring the *LogisticRegression models such that they extend the BaseModule.