Étape 1 : adapter votre script d'entraînement pour enregistrer un hook - Amazon SageMaker

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.

Étape 1 : adapter votre script d'entraînement pour enregistrer un hook

Amazon SageMaker Debugger est fourni avec une bibliothèque cliente appelée SDK sagemaker-debuggerPython. 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.