設定 DebuggerHookConfigAPI以儲存張量 - Amazon SageMaker

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

設定 DebuggerHookConfigAPI以儲存張量

使用 使用您在上一個步驟中建立的debugger_hook_config物件DebuggerHookConfigAPI來建立collection_configs物件。

from sagemaker.debugger import DebuggerHookConfig

debugger_hook_config=DebuggerHookConfig(
    collection_configs=collection_configs
)

Debugger 會將模型訓練輸出張量儲存至預設 S3 儲存貯體。預設 S3 儲存貯體的格式URI為 s3://sagemaker-<region>-<12digit_account_id>/<training-job-name>/debug-output/.

如果您想要指定確切的 S3 儲存貯體 URI，請使用下列程式碼範例：

from sagemaker.debugger import DebuggerHookConfig

debugger_hook_config=DebuggerHookConfig(
    s3_output_path="specify-amzn-s3-demo-bucket-uri"
    collection_configs=collection_configs
)

如需詳細資訊，請參閱 Amazon Python DebuggerHookConfig中的 。 SageMaker SDK