Налаштування тренера винагород
Ваш проєкт триває, і тепер у вас є об'єкти model і config, готові до початку навчання моделі винагороди.
Тренувальний і оцінювальний набори даних попередньо завантажено як train_data та eval_data. RewardTrainer імпортовано з trl.
Ця вправа є частиною курсу
Reinforcement Learning from Human Feedback (RLHF)
Інструкції до вправи
- Ініціалізуйте
RewardTrainer(), призначивши його атрибутам модель, токенайзер, тренувальний набір даних, оцінювальний набір даних і конфігурацію винагороди.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
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 = ____