CommencezCommencez gratuitement

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)

Voir le cours

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 = ____
Modifier et exécuter le code