Trainer

Trainer est une boucle complète d’entraînement et d’évaluation pour les modèles Transformers. Il vous suffit d’un modèle et d’un jeu de données pour commencer.

En interne, Trainer gère le lotage, le mélange et le bourrage de votre jeu de données en tenseurs. La boucle d’entraînement exécute le passage avant, calcule la perte, rétropropage les gradients et met à jour les poids. Configurez l’exécution de l’entraînement avec TrainingArguments pour personnaliser tout, de la taille du lot et de la durée de l’entraînement aux stratégies distribuées, à la compilation, et plus encore.

Étapes suivantes

• Commencez par le tutoriel sur le fine-tuning pour une introduction à l’entraînement d’un grand modèle de langage avec Trainer.
• Consultez le guide Méthodes de sous-classement de Trainer pour des exemples de sous-classement des méthodes de Trainer.
• Consultez le guide Collateurs de données pour apprendre à créer un collateur de données pour l’assemblage de lots personnalisés.
• Consultez le guide Callbacks pour apprendre à vous connecter aux événements d’entraînement.