本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
支持的框架
Amazon SageMaker Neo 支持以下框架。
框架 | 框架版本 | 模型版本 | 模型 | 模型格式(打包为 *.tar.gz) | 工具包 |
---|---|---|---|---|---|
MXNet | 1.6.0 | 支持 1.6.0 或更早版本 | 图像分类、对象检测、语义分割、姿势评估、活动识别 | 一个符号文件 (.json) 和一个参数文件 (.params) | GluonCV v0 |
ONNX | 1.5.0 | 支持 1.5.0 或更早版本 | 图像分类,SVM | 一个模型文件 (.onnx) | |
Keras | 2.2.4 | 支持 2.2.4 或更早版本 | 图像分类 | 一个模型定义文件 (h5) | |
PyTorch | 1.6.0 | 支持 1.6.0 或更早版本 | 图像分类 | 一个模型定义文件 (.pth) | |
TensorFlow | 1.15.0 | 支持 1.15.0 或更早版本 | 图像分类 | *对于保存的模型,一个 .pb 或一个 .pbtxt 文件,以及一个包含变量的变量目录 *对于冻结模型,仅一个 .pb 或 .pbtxt 文件 | |
TensorFlow-Lite | 1.13.1 | 支持 1.13.1 或更早版本 | 图像分类、对象检测 | 一个模型定义平面缓冲区文件 (.tflite) | |
XGBoost | 0.9 | 支持 0.9 或更早版本 | 决策树 | 一个 XGBoost 模型文件(.model),其中树中的节点数小于 2^31 | |
黑网 | 图像分类、对象检测 | 一个配置 (.cfg) 文件和一个权重 (.weights) 文件 |