JulesBelveze / bert-squeeze

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

feat: training assistant #38

Closed JulesBelveze closed 1 year ago

JulesBelveze commented 1 year ago

This PR aims at introducing a new helper object that makes the UX smoother: TrainAssistant. It directly loads a default configuration based on the model the user wants to fine-tune. It then instantiates the model, data, logger and callbacks.