Compilar un modelo (Amazon SageMaker Console)

Puede crear un trabajo de compilación de Amazon SageMaker Neo en la SageMaker consola de Amazon.

En la SageMaker consola de Amazon, selecciona Trabajos de compilación y, a continuación, selecciona Crear trabajo de compilación.
En la página Crear trabajo de compilación, en Nombre de trabajo, escriba un nombre. A continuación, selecciona un IAMrol.
Si no tiene un IAM rol, elija Crear un nuevo rol.
En la página Crear un IAM rol, elija Cualquier bucket de S3 y, a continuación, elija Crear rol.
Non PyTorch Frameworks

En la sección Configuración de entrada, introduzca la ruta completa del depósito de Amazon S3 URI que contiene los artefactos del modelo en el campo de entrada Ubicación de los artefactos del modelo. Los artefactos del modelo deben estar en un formato de archivo tar comprimido (.tar.gz).

En el campo de configuración de entrada de datos, introduzca la JSON cadena que especifica la forma de los datos de entrada.

Para Marco de machine learning, elija el marco que prefiera.

Para ver ejemplos de JSON cadenas de formas de datos de entrada que dependen de los marcos, consulta Qué formas de datos de entrada espera Neo.

PyTorch Framework
Se aplican instrucciones similares a la compilación de PyTorch modelos. Sin embargo, si ha entrenado con el modelo PyTorch y está intentando compilarlo para ml_* (exceptoml_inf) el objetivo, puede especificar, si lo desea, la versión PyTorch que ha utilizado.

Para ver ejemplos de JSON cadenas de formas de datos de entrada que dependen de los marcos, consulta Qué formas de datos de entrada espera Neo.
Notas

Si guardó el modelo con la PyTorch versión 2.0 o posterior, el campo de configuración de entrada de datos es opcional. SageMaker Neo obtiene la configuración de entrada del archivo de definición del modelo con el que se crea PyTorch. Para obtener más información sobre cómo crear el archivo de definición, consulte la PyTorch sección sobre cómo guardar modelos para SageMaker Neo.

Al compilar ml_* instancias mediante un PyTorch framework, utilice el campo de opciones del compilador en la configuración de salida para proporcionar el tipo de datos correcto (dtype) de la entrada del modelo. El valor predeterminado se establece en "float32".
aviso
Si especificas una URI ruta de bucket de Amazon S3 que lleva al .pth archivo, recibirás el siguiente error después de iniciar la compilación: ClientError: InputConfiguration: Unable to untar input model.Please confirm the model is a tar.gz file
Vaya a la sección de Configuración de salida. Elija dónde desea implementar su modelo. Puede implementar su modelo en un Dispositivo de destino o en una Plataforma de destino. Los dispositivos de destino incluyen dispositivos periféricos y en la nube. Las plataformas de destino hacen referencia a sistemas operativos, arquitectura y aceleradores específicos en los que desea que se ejecute su modelo.

En Ubicación de salida de S3, escriba la ruta o al bucket de S3 o donde desee almacenar el modelo. Si lo desea, puede añadir opciones del compilador en JSON formato en la sección Opciones del compilador.
Compruebe el estado del trabajo de compilación cuando se inicie. Este estado del trabajo se encuentra en la parte superior de la página del Trabajo de compilación, como se muestra en la siguiente captura de pantalla. También puede comprobar su estado en la columna Estado.
Compruebe el estado del trabajo de compilación cuando se haya completado. Puede comprobar el estado en la columna Estado, como se muestra en la siguiente captura de pantalla.