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

Amazon SageMaker AI 如何提供訓練資訊

本節說明 SageMaker AI 如何讓訓練資訊可供您的 Docker 容器使用，例如訓練資料、超參數和其他組態資訊。

當您傳送CreateTrainingJob請求至 SageMaker AI 以開始模型訓練時，您可以指定包含訓練演算法之 Docker 映像的 Amazon Elastic Container Registry (Amazon ECR) 路徑。您亦需指定 Amazon Simple Storage Service (Amazon S3) 的位置，其為訓練資料和演算法專屬參數的存放位置。SageMaker AI 可讓 Docker 容器使用此資訊，讓您的訓練演算法可以使用它。本節說明我們如何讓這項資訊可供 Docker 容器使用。如需建立訓練工作的資訊，請參閱 CreateTrainingJob。如需 SageMaker AI 容器組織資訊方式的詳細資訊，請參閱 SageMaker 訓練和推論工具組。

超參數

SageMaker AI 可讓CreateTrainingJob請求中的超參數在 /opt/ml/input/config/hyperparameters.json 檔案的 Docker 容器中可用。

以下是 hyperparameters.json 中的超參數組態範例，用來指定在 CreateTrainingJob 作業中 XGBoost 的 num_round 和 eta 超參數。

{
    "num_round": "128",
    "eta": "0.001"
}            

如需可用於 SageMaker AI 內建 XGBoost 演算法的超參數完整清單，請參閱 XGBoost 超參數。

您可以調整的超參數，依您正在訓練的演算法而定。如需適用於 SageMaker AI 內建演算法的超參數清單，請在使用 Amazon SageMaker AI 內建演算法或預先訓練模型的演算法連結下，在超參數中尋找這些項目。

環境變數

SageMaker AI 會在您的容器中設定下列環境變數：

輸入資料組態

SageMaker AI 可讓您的CreateTrainingJob請求在 Docker 容器中的 /opt/ml/input/config/inputdataconfig.json 檔案中提供 InputDataConfig 參數中的資料頻道資訊。

舉例來說，假設您在請求中指定三個資料通道 (train、evaluation 與 validation)，SageMaker AI 提供下列 JSON：

{
  "train" : {"ContentType":  "trainingContentType",
             "TrainingInputMode": "File",
             "S3DistributionType": "FullyReplicated",
             "RecordWrapperType": "None"},
  "evaluation" : {"ContentType":  "evalContentType",
                  "TrainingInputMode": "File",
                  "S3DistributionType": "FullyReplicated",
                  "RecordWrapperType": "None"},
  "validation" : {"TrainingInputMode": "File",
                  "S3DistributionType": "FullyReplicated",
                  "RecordWrapperType": "None"}
} 

訓練資料

CreateTrainingJob 請求中 AlgorithmSpecification 的 TrainingInputMode 參數會指定如何將訓練資料集提供給您的容器。以下是可用的輸入模式：

SageMaker AI 模型訓練支援高效能 S3 Express One Zone 目錄儲存貯體，做為檔案模式、快速檔案模式和管道模式的資料輸入位置。若要使用 S3 Express One Zone，請輸入 S3 Express One Zone 目錄儲存貯體的位置，而非 Amazon S3 一般用途儲存貯體。為 IAM 角色提供 ARN 與必要的存取控制和許可政策。如需詳細資訊，請參閱 AmazonSageMakerFullAccesspolicy。您只能使用 Amazon S3 受管金鑰 (SSE-S3) 加密伺服器端加密目錄儲存貯體中的 SageMaker AI 輸出資料。目前不支援使用 AWS KMS 金鑰 (SSE-KMS) 的伺服器端加密，以將 SageMaker AI 輸出資料儲存在目錄儲存貯體中。如需詳細資訊，請參閱 S3 Express One Zone。

分散式訓練組態

如果您使用多個容器執行分散式訓練，SageMaker AI 會提供有關/opt/ml/input/config/resourceconfig.json檔案中所有容器的資訊。

此 JSON 檔案包含所有容器的資訊，讓您可以啟用容器間的通訊。SageMaker AI 讓此檔案同時適用於 File和 Pipe 模式演算法。該檔案提供下列資訊：

以下是三節點叢集的第 1 節點範例檔案：

{
    "current_host": "algo-1",
    "hosts": ["algo-1","algo-2","algo-3"],
    "network_interface_name":"eth1"
}