Configurer l'DebuggerHookConfigAPI pour enregistrer les tenseurs - 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.

Configurer l'DebuggerHookConfigAPI pour enregistrer les tenseurs

Utilisez l'DebuggerHookConfigAPI pour créer un debugger_hook_config objet à l'aide de l'collection_configsobjet que vous avez créé à l'étape précédente.

from sagemaker.debugger import DebuggerHookConfig

debugger_hook_config=DebuggerHookConfig(
    collection_configs=collection_configs
)

Debugger enregistre les tenseurs de sortie d'entraînement du modèle dans le compartiment S3 par défaut. 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/.

Si vous souhaitez spécifier un URI de compartiment S3 précis, utilisez l'exemple de code suivant :

from sagemaker.debugger import DebuggerHookConfig

debugger_hook_config=DebuggerHookConfig(
    s3_output_path="specify-uri"
    collection_configs=collection_configs
)

Pour plus d'informations, consultez DebuggerHookConfigle SDK Amazon SageMaker Python.