Framework supportati - Amazon SageMaker AI

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

Framework supportati

Amazon SageMaker Neo supporta i seguenti framework.

Framework Versione di Framework Versioni del modello Modelli Formati di modelli (pacchettizzati in *.tar.gz) Kit di strumenti
MXNet 1.8 Supporta 1.8 o versioni precedenti Classificazione delle immagini, Rilevamento di oggetti, Segmentazione semantica, Stima della posa, Riconoscimento delle attività Un file di simboli (.json) e un file di parametri (.params) GluonCV versione 0.8.0
ONNX 1,7 Supporta 1.7 o versioni precedenti Classificazione immagine, SVM Un file di modello (.onnx)
Keras 2.2 Supporta 2.2 o versioni precedenti Classificazione delle immagini Un file di definizione del modello (.h5)
PyTorch 1.7, 1.8 Supporta 1.7, 1.8 o versioni precedenti Classificazione delle immagini, Rilevamento di oggetti Un file di definizione del modello (.pth)
TensorFlow 1.15, 2.4, 2.5 (solo per le istanze ml.inf1.*) Supporta 1.15, 2.4, 2.5 (solo per istanze ml.inf1.*) o versioni precedenti Classificazione delle immagini, Rilevamento di oggetti *Per i modelli salvati, un file .pb o uno .pbtxt e una directory delle variabili che contiene le variabili *Per i modelli congelati, solo un file .pb o .pbtxt
TensorFlow-Leggero 1.15 Supporta 1.15 o versioni precedenti Classificazione delle immagini, Rilevamento di oggetti Un file flatbuffer di definizione del modello (.tflite)
XGBoost 1.3 Supporta 1.3 o versioni precedenti Alberi decisionali Un file di XGBoost modello (.model) in cui il numero di nodi in un albero è inferiore a 2^31
DARKNET Classificazione delle immagini, Rilevamento degli oggetti (il modello Yolo non è supportato) Un file di configurazione (.cfg) e un file di pesi (.weights)