rajpurkar/squad
Viewer • Updated • 98.2k • 150k • 363
How to use philschmid/distilbert-onnx with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="philschmid/distilbert-onnx") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("philschmid/distilbert-onnx")
model = AutoModelForQuestionAnswering.from_pretrained("philschmid/distilbert-onnx")This model is a fine-tune checkpoint of DistilBERT-base-cased, fine-tuned using (a second step of) knowledge distillation on SQuAD v1.1. This model reaches a F1 score of 87.1 on the dev set (for comparison, BERT bert-base-cased version reaches a F1 score of 88.7).