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 L'IA stocke votre modèle /opt/ml/model et vos données dedans/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 de formation en mode de téléchargement non compressé en utilisant le AWS Command Line Interface (AWS CLI) ou le SDK SageMaker Python.

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 de formation en mode de téléchargement non compressé à l'aide du SDK SageMaker Python.

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
)