Cadres pris en charge

Amazon SageMaker Neo prend en charge les cadres suivants.

Cadre	Version du cadre	Version de modèle	Modèles	Formats de modèle (packagés dans *.tar.gz)	Boîtes à outils
MXNet	1.8	Prend en charge la version 1.8 ou antérieure	classification d'images, détection d'objets, segmentation sémantique, estimation de pose, reconnaissance d'activités	Un fichier de symboles (.json) et un fichier de paramètres (.params)	GluonCV v0.8.0
ONNX	1,7	Prend en charge la version 1.7 ou antérieure	Classification d'images, SVM	Un fichier de modèle (.onnx)
Keras	2.2	Prend en charge la version 2.2 ou antérieure	Classification d'images	Un fichier de définition de modèle (.h5)
PyTorch	1.7, 1.8	Prend en charge la version 1.7, 1.8 ou antérieure	Classification d'images, détection d'objets	Un fichier de définition de modèle (.pth)
TensorFlow	1.15, 2.4, 2.5 (uniquement pour les instances ml.inf1.*)	Prend en charge les versions 1.15, 2.4, 2.5 (uniquement pour les instances ml.inf1.*) ou antérieures	Classification d'images, détection d'objets	*Pour les modèles enregistrés, un fichier .pb ou .pbtxt et un répertoire de variables contenant des variables *Pour les modèles figés, un seul fichier .pb ou .pbtxt
TensorFlow-Lite	1.15	Prend en charge la version 1.15 ou antérieure	Classification d'images, détection d'objets	Un fichier de tampon plat de définition de modèle (.tflite)
XGBoost	1.3	Prend en charge la version 1.3 ou antérieure	Arbres de décision	Un fichier de modèles XGBoost (.model) dans lequel le nombre de nœuds d'une arborescence est inférieur à 2^31
DARKNET			Classification des images, détection d'objets (le modèle Yolo n'est pas pris en charge)	Un fichier de configuration (.cfg) et un fichier de poids (.weights)