feat(prediction): add classes for inference

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

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

79 stars 10 forks source link

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

This PR aims at providing a model wrapper in order to perform inference.