Configuration du SageMaker débogueur pour enregistrer les tenseurs

Les tenseurs sont des ensembles de données de paramètres mis à jour à partir des étapes aller-retour de chaque itération d'entraînement. SageMaker Debugger collecte les tenseurs de sortie pour analyser l'état d'une tâche de formation. SageMaker Les opérations du débogueur CollectionConfiget de DebuggerHookConfigl'API fournissent des méthodes pour regrouper les tenseurs en collections et les enregistrer dans un compartiment S3 cible. Les rubriques suivantes montrent comment utiliser les opérations CollectionConfig et DebuggerHookConfig API, suivies d'exemples d'utilisation du hook Debugger pour enregistrer, accéder et visualiser les tenseurs de sortie.

Lors de la construction d'un estimateur SageMaker AI, activez SageMaker Debugger en spécifiant le paramètre. debugger_hook_config Les rubriques suivantes incluent des exemples de configuration des opérations d'debugger_hook_configutilisation CollectionConfig et d'DebuggerHookConfigAPI pour extraire les tenseurs de vos tâches de formation et les enregistrer.

Note

Une fois correctement configuré et activé, SageMaker Debugger enregistre les tenseurs de sortie dans un compartiment S3 par défaut, sauf indication contraire. Le format de l'URI du compartiment S3 par défaut est s3://amzn-s3-demo-bucket-sagemaker-<region>-<12digit_account_id>/<training-job-name>/debug-output/.

Rubriques

• Configurer les collections de tenseurs à l'aide de l'API CollectionConfig

• Configurer l'DebuggerHookConfigAPI pour enregistrer les tenseurs

• Exemples de blocs-notes et d'exemples de code pour configurer Debugger Hook