本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。
支援的架構
Amazon SageMaker 支援以下架構。
架構 | 框架版本 | 模型版本 | 模型 | 模型格式 (以 *.tar.gz 封裝) | 工具組 |
---|---|---|---|---|---|
MXNet | 1.8 | 支援 1.8 或更早版本 | 影像分類、物件偵測、語意分割、姿勢估算、活動辨識 | 一個符號檔案 (.json) 和一個參數檔案 (.params) | GluonCV v0.8.0 |
ONNX | 1.7 | 支援 1.7 或更早版本 | 影像影像分類、SVM | 一個模型檔案 (.onnx) | |
Keras | 2.2 | 支援 2.2 或更早版本 | 影像分類 | 一個模型定義檔案 (.h5) | |
PyTorch | 1.7、1.8 | 支援 1.7、1.8 或更早版本 | 影像分類、物件偵測 | 一個模型定義檔案 (.pth) | |
TensorFlow | 1.15、2.4、2.5 (僅適用於 ml.inf1.* 執行個體) | 支援 1.15、2.4、2.5 (僅適用於 ml.inf1 執行個體) 或更早版本 | 影像分類、物件偵測 | *對於儲存模型,一個 .pb 或一個 .pbtxt 檔案和一個包含變數的變數目錄 *對於凍結的模型,只有一個 .pb 或 .pbtxt 檔案 | |
TensorFlow 精簡版 | 1.15 | 支援 1.15 或更早版本 | 影像分類、物件偵測 | 一個模型定義 Flatbuffer 檔案 (.tflite) | |
XGBoost | 1.3 | 支援 1.3 或更早版本 | 決策樹 | 一個 XGBoost 模型檔案 (模型),其中樹中的節點數量低於 2^31 | |
DARKNET | 影像分類、物件偵測 (不支援 Yolo 模型) | 一個組態 (.cfg) 檔案和一個權重 (.weights) 檔案 |