Configuration de l’API DebuggerHookConfig 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.

Configuration de l’API DebuggerHookConfig pour enregistrer les tenseurs

Utilisez l'API DebuggerHookConfig pour créer un objet debugger_hook_config à l'aide de l'objet collection_configs 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 DebuggerHookConfig dans Amazon SageMaker Python SDK.