CommencezCommencez gratuitement

Initialiser la récompense

Vous êtes aux dernières étapes du déploiement d'un modèle génératif conçu pour offrir des recommandations personnalisées pour une librairie en ligne. Pour que ce modèle corresponde aux recommandations préférées par les personnes utilisatrices, vous devez entraîner un modèle de récompense à partir de données de préférences recueillies. La première étape consiste à initialiser le modèle et les paramètres de configuration.

Les objets AutoTokenizer et AutoModelForSequenceClassification ont été préchargés depuis transformers. RewardConfig a été préchargé depuis trl.

Cette activité fait partie du cours

Reinforcement Learning from Human Feedback (RLHF)

Voir le cours

Instructions de l’exercice

  • Chargez le modèle GPT-1, "openai-gpt", pour la tâche de classification de séquences en utilisant AutoModelForSequenceClassification de Hugging Face.
  • Initialisez la configuration de la récompense en utilisant "output_dir" comme répertoire de sortie et réglez la longueur maximale des jetons à 60.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Load the pre-trained GPT-1 model for text classification
model = ____

tokenizer = AutoTokenizer.from_pretrained("openai-gpt")

# Initialize the reward configuration and set max_length
config = ____(output_dir=____, max_length=____)
Modifier et exécuter le code