Konfigurera träningshanteraren
Nu när träningsargumenten är konfigurerade är det dags att sätta upp träningshanteraren innan du kan använda den finjusterade modellen.
TrainingArguments och Trainer har redan laddats åt dig, tillsammans med din tidigare model, tokenizer och training_args.
Obs! Den faktiska träningen har inaktiverats i den här övningen för att den ska gå snabbare.
Den här övningen är en del av kursen
Introduktion till LLM:er i Python
Övningsinstruktioner
- Skapa
Trainer()-objektet. - Tilldela de tidigare definierade träningsargumenten och tokeniseraren.
- Träna modellen.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Set up the trainer object
trainer = ____(
model=model,
# Assign the training arguments and tokenizer
____,
train_dataset=tokenized_training_data,
eval_dataset=tokenized_test_data,
____
)
# Train the model
____