I\'m using pytorch and this is the model from huggingface transformers https://huggingface.co/transformers/model_doc/bert.html#bertforsequenceclassification: