Kerangka Kerja yang Didukung - Amazon SageMaker

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Kerangka Kerja yang Didukung

Amazon SageMaker Neo mendukung kerangka kerja berikut.

Kerangka Kerja Framework Versi Versi Model Model Format Model (dikemas dalam *.tar.gz) Toolkit
MXNet 1.8 Mendukung 1.8 atau yang lebih baru Klasifikasi Gambar, Deteksi Objek, Segmentasi Semantik, Estimasi Pose, Pengenalan Aktivitas Satu file simbol (.json) dan satu file parameter (.params) GluonCV
ONNX 1.7 Mendukung 1.7 atau yang lebih baru Klasifikasi Citra Citra, SVM Satu file model (.onnx)
Keras 2.2 Mendukung 2.2 atau yang lebih baru Klasifikasi Citra Satu file definisi model (.h5)
PyTorch 1.7, 1,8 Mendukung 1.7, 1.8 atau sebelumnya Klasifikasi Gambar, Deteksi Objek Satu file definisi model (.pth)
TensorFlow 1.15, 2.4, 2.5 (hanya untuk instans ml.inf1.*) Mendukung 1.15, 2.4, 2.5 (hanya untuk instans ml.inf1.*) atau sebelumnya Klasifikasi Gambar, Deteksi Objek * Untuk model yang disimpan, satu file.pb atau satu file.pbtxt dan direktori variabel yang berisi variabel*Untuk model beku, hanya satu file.pb atau .pbtxt
TensorFlow-Ringan 1.15 Mendukung 1.15 atau yang lebih baru Klasifikasi Gambar, Deteksi Objek Satu model definisi flatbuffer file (.tflite)
XGBoost 1.3 Mendukung 1.3 atau yang lebih baru Pohon Keputusan Satu file model XGBoost (.model) di mana jumlah node dalam pohon kurang dari 2 ^ 31
DARKNET Klasifikasi Gambar, Deteksi Objek (model Yolo tidak didukung) Satu file konfigurasi (.cfg) dan satu file bobot (.weights)