Bắt đầu ngayBắt đầu miễn phí

Thiết lập reward trainer

Dự án của bạn tiếp tục và hiện bạn đã có các đối tượng modelconfig sẵn sàng để bắt đầu huấn luyện reward model.

Các tập dữ liệu huấn luyện và đánh giá đã được nạp sẵn thành train_dataeval_data. RewardTrainer đã được import từ trl.

Bài tập này là một phần của khóa học

Reinforcement Learning from Human Feedback (RLHF)

Xem khóa học

Hướng dẫn bài tập

  • Khởi tạo RewardTrainer() bằng cách gán model, tokenizer, tập dữ liệu huấn luyện, tập dữ liệu đánh giá và cấu hình phần thưởng vào các thuộc tính tương ứng của nó.

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.

tokenizer = AutoTokenizer.from_pretrained("openai-gpt")
model = AutoModelForSequenceClassification.from_pretrained('openai-gpt')
config = RewardConfig(output_dir='output_dir', max_length=60)

# Initialize the reward trainer
trainer = ____
Chỉnh sửa và Chạy Mã