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á.
SageMaker O Amazon Neo oferece suporte às seguintes estruturas.
| Framework | Versão do framework | Versão do modelo | Modelos da | 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 XGBoost modelo (.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) |
