Thiết lập trainer
Sau khi đã thiết lập các tham số huấn luyện, bạn cần chuyển sang cấu hình trainer trước khi đưa mô hình fine-tuned vào sử dụng.
TrainingArguments và Trainer đã được nạp sẵn cho bạn, cùng với model, tokenizer và training_args trước đó.
Lưu ý: việc huấn luyện thực tế đã được tắt trong bài tập này để bài chạy nhanh hơn.
Bài tập này là một phần của khóa học
Nhập môn LLMs trong Python
Hướng dẫn bài tập
- Thiết lập đối tượng
Trainer(). - Gán các tham số huấn luyện và tokenizer đã định nghĩa trước đó.
- Huấn luyện mô hình.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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
____