A text classifier made for people in a hurry
This project aims at simplifying training and productionizing of BERT models easier for everyone without delving into details.
Uses PyTorch Lightning module along with support for GPU training and inference
- Bug fixes and ironing out issues in training
- Inference script
- Containerization