Configurazione dell’API DebuggerHookConfig per il salvataggio dei tensori - Amazon SageMaker AI

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

Configurazione dell’API DebuggerHookConfig per il salvataggio dei tensori

Utilizza l'API DebuggerHookConfig per creare un oggetto debugger_hook_config utilizzando l'oggetto collection_configs creato nella fase precedente.

from sagemaker.debugger import DebuggerHookConfig

debugger_hook_config=DebuggerHookConfig(
    collection_configs=collection_configs
)

Debugger salva i tensori di output di addestramento del modello nel bucket S3 predefinito. Il formato dell'URI del bucket S3 predefinito è s3://amzn-s3-demo-bucket-sagemaker-<region>-<12digit_account_id>/<training-job-name>/debug-output/.

Se desideri specificare un URI esatto del bucket S3, usa il seguente esempio di codice:

from sagemaker.debugger import DebuggerHookConfig

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

Per ulteriori informazioni, consulta DebuggerHookConfig in Amazon SageMaker Python SDK.