Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.
Marcos admitidos
Amazon SageMaker Neo es compatible con los siguientes marcos.
| Marcos | Versión de marco | Versión del modelo | Modelos | Formatos de modelo (empaquetados en *.tar.gz) | Kits de herramientas |
|---|---|---|---|---|---|
| MXNet | 1.8 | Compatible con 1.8 o versiones anteriores | Clasificación de imágenes, detección de objetos, segmentación semántica, estimación de poses, reconocimiento de actividades | Un archivo de símbolos (.json) y un archivo de parámetros (.params) | GluonCV v0.8.0 |
| ONNX | 1.7 | Compatible con 1.7 o versiones anteriores | Clasificación de imágenes, SVM | Un archivo de modelos (.onnx) | |
| Keras | 2.2 | Compatible con 2.2 o versiones anteriores | Clasificación de imágenes | Un archivo de definición de modelo (.h5) | |
| PyTorch | 1.7, 1.8 | Compatible con 1.7, 1.8 o versiones anteriores | Clasificación de imágenes, detección de objetos | Un archivo de definición de modelo (.pth) | |
| TensorFlow | 1.15, 2.4, 2.5 (solo para instancias ml.inf1.*) | Compatible con 1.15, 2.4, 2.5 (solo para instancias ml.inf1.*) o versiones anteriores | Clasificación de imágenes, detección de objetos | *En el caso de los modelos guardados, un archivo .pb o .pbtxt y un directorio de variables que contiene variables. *En el caso de modelos congelados, solo un archivo .pb o .pbtxt | |
| TensorFlow-Lite | 1.15 | Compatible con 1.15 o versiones anteriores | Clasificación de imágenes, detección de objetos | Un archivo de búfer plano de definición de modelo (.tflite) | |
| XGBoost | 1.3 | Compatible con 1.3 o versiones anteriores | Árboles de decisión | Un archivo de modelo de XGBoost (.model) en el que el número de nodos de un árbol es inferior a 2^31 | |
| DARKNET | Clasificación de imágenes, detección de objetos (el modelo Yolo no es compatible) | Un archivo de configuración (.cfg) y un archivo de pesos (.weights) |
