Sortie de modèle non compressée - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Sortie de modèle non compressée

SageMaker AI stocke votre modèle dans /opt/ml/model et vos données dans /opt/ml/output/data. Une fois le modèle et les données écrits dans ces emplacements, ils sont chargés dans votre compartiment Amazon S3 sous forme de fichiers compressés par défaut.

Vous pouvez gagner du temps sur la compression de fichiers de données volumineux en téléchargeant le modèle et les sorties de données dans votre compartiment S3 sous forme de fichiers non compressés. Pour ce faire, créez une tâche d'entraînement en mode de téléchargement non compressé à l'aide de l'AWS Command Line Interface (AWS CLI) ou du kit SageMaker Python SDK.

L'exemple de code suivant montre comment créer une tâche d'entraînement en mode chargement non compressé lorsque vous utilisez AWS CLI. Pour activer le mode de téléchargement non compressé, définissez le champ CompressionType dans l'API OutputDataConfig sur NONE.

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

L'exemple de code suivant montre comment créer une tâche d'entraînement en mode téléchargement non compressé à l'aide du kit 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
)