Menyiapkan trainer
Setelah argumen pelatihan disiapkan, Anda perlu beralih ke trainer sebelum menggunakan model yang telah di-fine-tune.
TrainingArguments dan Trainer telah dimuat untuk Anda, begitu juga model, tokenizer, dan training_args sebelumnya.
Catatan: pelatihan aktual telah dinonaktifkan untuk latihan ini agar proses berjalan lebih cepat.
Latihan ini adalah bagian dari kursus
Pengantar LLM di Python
Petunjuk latihan
- Siapkan objek
Trainer(). - Gunakan argumen pelatihan dan tokenizer yang telah didefinisikan sebelumnya.
- Latih model.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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
____