Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Load DistilBERT with [AutoModelForQuestionAnswering]:
from transformers import AutoModelForQuestionAnswering, TrainingArguments, Trainer
model = AutoModelForQuestionAnswering.from_pretrained("distilbert/distilbert-base-uncased")
At this point, only three steps remain:
Define your training hyperparameters in [TrainingArguments].