Konfiguracja trenera
Po skonfigurowaniu argumentów trenowania czas przejść do trenera, a następnie użyć wytrenowanego modelu.
TrainingArguments i Trainer zostały już załadowane, podobnie jak wcześniej zdefiniowane: model, tokenizer i training_args.
Uwaga: faktyczne trenowanie zostało wyłączone w tym ćwiczeniu, żeby działało szybciej.
To ćwiczenie jest częścią kursu
Wprowadzenie do LLM w Pythonie
Instrukcje do ćwiczenia
- Skonfiguruj obiekt
Trainer(). - Przypisz wcześniej zdefiniowane argumenty trenowania oraz tokenizer.
- Wytrenuj model.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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
____