View a markdown version of this page

Adaptation de votre script d’entraînement pour enregistrer un hook - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Adaptation de votre script d’entraînement pour enregistrer un hook

Note

Amazon SageMaker Debugger n'est plus ouvert aux nouveaux clients. Les clients existants peuvent continuer à utiliser le service normalement. AWS continue d'investir dans l'amélioration de la sécurité et de la disponibilité de Debugger, mais nous ne prévoyons pas d'introduire de nouvelles fonctionnalités. Pour de plus amples informations, veuillez consulter Modification de la disponibilité du débogueur.

Amazon SageMaker Debugger est fourni avec une bibliothèque cliente appelée sagemaker-debugger Python SDK. Le kit SDK Python pour sagemaker-debugger fournit des outils pour adapter votre script d'entraînement avant l'entraînement et des outils d'analyse après entraînement. Sur cette page, vous allez apprendre à adapter votre script d'entraînement à l'aide de la bibliothèque client.

Le kit SDK Python pour sagemaker-debugger fournit des fonctions de wrapper qui aident à enregistrer un hook pour extraire les tenseurs du modèle, sans altérer votre script d'entraînement. Pour commencer à collecter les tenseurs de sortie du modèle et à les déboguer pour trouver les problèmes d'entraînement, apportez les modifications suivantes à votre script d'entraînement.

Astuce

Pendant que vous suivez cette page, utilisez la documentation sur le kit SDK open source pour sagemaker-debugger pour accéder aux références d'API.