Configurer le formateur de récompense
Votre projet se poursuit et vous avez maintenant les objets model et config prêts pour entraîner le modèle de récompense.
Les jeux de données d'entraînement et d'évaluation ont été préchargés sous train_data et eval_data. RewardTrainer a été importé de trl.
Cette activité fait partie du cours
Reinforcement Learning from Human Feedback (RLHF)
Instructions de l’exercice
- Initialisez
RewardTrainer()en attribuant le modèle, le tokenizer, le jeu de données d'entraînement, le jeu de données d'évaluation et la configuration de récompense à ses attributs.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
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 = ____