ตั้งค่า Trainer
เมื่อกำหนด training arguments เรียบร้อยแล้ว ขั้นตอนต่อไปคือตั้งค่า Trainer ก่อนนำโมเดลที่ fine-tune แล้วไปใช้งาน
TrainingArguments และ Trainer ถูกโหลดให้แล้ว พร้อมกับ model, tokenizer และ training_args จากขั้นตอนก่อนหน้า
หมายเหตุ: การฝึกจริงถูกปิดใช้งานในแบบฝึกหัดนี้เพื่อให้รันได้เร็วขึ้น
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Python เบื้องต้นสำหรับ LLMs
คำแนะนำการฝึกหัด
- สร้างออบเจกต์
Trainer() - กำหนด training arguments และ tokenizer ที่กำหนดไว้ก่อนหน้า
- ฝึกโมเดล
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
# 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
____