支援的架構

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) 檔案