search

JulesBelveze / bert-squeeze

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

fix(bertology): access to leaf tensor #18

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

This PR aims at fixing the LotteryTicket callback and enable to use other metrics than just accuracy.