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