Estruturas compatíveis - Amazon SageMaker

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Estruturas compatíveis

SageMaker O Amazon Neo oferece suporte às seguintes estruturas.

Framework Versão da estrutura Versão do modelo Modelos Formatos de modelo (empacotados em *.tar.gz) Kits de ferramentas
MXNet 1.8 Compatível com 1.8 ou anterior Classificação de imagens, detecção de objetos, segmentação semântica, estimativa de pose, reconhecimento de atividades Um arquivo de símbolos (.json) e um arquivo de parâmetros (.params) GluonCV v0.8.0
ONNX 1,7 Compatível com 1.7 ou anterior Classificação de imagens, SVM Um arquivo de modelo (.onnx)
Keras 2.2 Compatível com 2.2 ou anterior Classificação de imagens Um arquivo de definição de modelo (.h5)
PyTorch 1,7, 1,8 Compatível com 1.7, 1.8 ou anterior Classificação de imagens, detecção de objetos Um arquivo de definição de modelo (.pth)
TensorFlow 1,15, 2,4, 2,5 (somente para instâncias ml.inf1.*) Compatível com instâncias 1.15, 2.4, 2.5 ou anteriores (somente para instâncias ml.inf1.*) Classificação de imagens, detecção de objetos *Para modelos salvos, um arquivo .pb ou um arquivo .pbtxt e um diretório de variáveis que contenha variáveis *Para modelos congelados, apenas um arquivo .pb ou .pbtxt
TensorFlow-Leve 1.15 Compatível com 1.15 ou anterior Classificação de imagens, detecção de objetos Um arquivo flatbuffer de definição de modelo (.tflite)
XGBoost 1.3 Compatível com 1.3 ou anterior Árvores de decisão Um arquivo de modelo XGBoost (.model) em que o número de nós em uma árvore é menor que 2^31
DARKNET Classificação de imagens, detecção de objetos (o modelo Yolo não é compatível) Um arquivo de configuração (.cfg) e um arquivo de pesos (.weights)