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 model và config 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_data và eval_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)
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 = ____