架構分析的估算器組態 - Amazon SageMaker

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

架構分析的估算器組態

警告

為了有利於 Amazon SageMaker Profiler, SageMakerDebugger 從 TensorFlow 2.11 和 PyTorch 2.0 開始取代架構分析功能。您仍然可以在舊版的架構和 中使用此功能SDKs,如下所示。

  • SageMaker Python SDK <= v2.130.0

  • PyTorch >= v1.6.0、< v2.0

  • TensorFlow >= v2.3.1、< v2.11

另請參閱2023 年 3 月 16 日

若要啟用偵錯工具架構設定,請在建構估算器時設定 framework_profile_params 參數。Debugger 架構分析會收集架構指標,例如初始化階段的資料、資料載入器程序、深度學習架構和訓練指令碼的 Python 運算子、步驟內和步驟之間的詳細分析,以及 cProfile 或 Pyinstrument 選項。使用 FrameworkProfile 類別,您可以設定自訂架構分析選項。

注意

在開始使用偵錯工具架構分析之前,請先驗證偵錯工具是否支援用於建立模型的框架進行框架分析。如需詳細資訊,請參閱支援的架構和演算法

偵錯工具會將架構指標儲存在預設 S3 儲存貯體中。預設 S3 儲存貯體的格式URI為 s3://sagemaker-<region>-<12digit_account_id>/<training-job-name>/profiler-output/