未壓縮模型輸出 - Amazon SageMaker

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

未壓縮模型輸出

SageMaker 會將您的模型儲存在 中,/opt/ml/model並將資料儲存在 中/opt/ml/output/data。將模型和資料寫入這些位置後,根據預設以壓縮檔案形式上傳到 Amazon S3 儲存貯體。

您可以以未壓縮檔案形式將模型和資料輸出上傳到 S3 儲存貯體,節省大型資料檔案壓縮的時間。若要這麼做,請使用 AWS Command Line Interface (AWS CLI) 或 SageMaker Python ,在未壓縮上傳模式下建立訓練任務SDK。

下列程式碼範例示範如何在使用 AWS CLI時在未壓縮上傳模式下建立訓練任務。若要啟用未壓縮上傳模式,請將 中的CompressionType欄位設定為 OutputDataConfig API NONE

{ "TrainingJobName": "uncompressed_model_upload", ... "OutputDataConfig": { "S3OutputPath": "s3://amzn-s3-demo-bucket/uncompressed_upload/output", "CompressionType": "NONE" }, ... }

下列程式碼範例示範如何使用 SageMaker Python 以未壓縮的上傳模式建立訓練任務SDK。

import sagemaker from sagemaker.estimator import Estimator estimator = Estimator( image_uri="your-own-image-uri", role=sagemaker.get_execution_role(), sagemaker_session=sagemaker.Session(), instance_count=1, instance_type='ml.c4.xlarge', disable_output_compression=True )