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)
Instructions de l’exercice
- Chargez le modèle GPT-1,
"openai-gpt", pour la tâche de classification de séquences en utilisantAutoModelForSequenceClassificationde 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=____)