Frameworks pris en charge - Amazon SageMaker

Frameworks pris en charge

Amazon SageMaker Neo prend en charge les cadres suivants.

Infrastructure Version Framework Version de modèle Modèles Formats de modèle (packagés dans *.tar.gz) Boîtes à outils
MXNet 1.8.0 Prend en charge la version 1.8.0 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.0 Prend en charge la version 1.7.0 ou antérieure Classification d'images, SVM Un fichier de modèle (.onnx)
Keras 2.2.4 Prend en charge la version 2.2.4 ou antérieure Classification d'images Un fichier de définition de modèle (.h5)
PyTorch 1.7.1, 1.8.1 Prend en charge les versions 1.7.1, 1.8.1 ou antérieures Classification d'images, détection d'objets Un fichier de définition de modèle (.pth)
TensorFlow 1.15.0, 2.4.2 Prend en charge les versions 1.15.0, 2.4.2 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.2 Prend en charge la version 1.15.2 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.3 Prend en charge la version 1.3.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 d'images, détection d'objets Un fichier de configuration (.cfg) et un fichier de poids (.weights)