ПочатиПочніть безкоштовно

Налаштування тренера винагород

Ваш проєкт триває, і тепер у вас є об'єкти 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 = ____
Редагувати та запускати код