pytorch-transformers-classification
ThilinaRajapakse/pytorch-transformers-classification
Jupyter Notebook
Based on the Pytorch-Transformers library by HuggingFace. To be used as a starting point for employing Transformer models in text classification tasks. Contains code to easily train BERT, XLNet, RoBERTa, and XLM models for text classification.