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 Versi Kerangka Versi Model Model Format Model (dikemas dalam*.tar.gz) Toolkit
MXNet 1.8 Mendukung 1,8 atau sebelumnya Klasifikasi Gambar, Deteksi Objek, Segmentasi Semantik, Estimasi Pose, Pengenalan Aktivitas Satu file simbol (.json) dan satu file parameter (.params) GluonCV v0.8.0
ONNX 1.7 Mendukung 1.7 atau sebelumnya Klasifikasi Gambar, SVM Satu file model (.onnx)
Keras 2.2 Mendukung 2.2 atau sebelumnya Klasifikasi Gambar 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 instance ml.inf1.*) Mendukung 1.15, 2.4, 2.5 (hanya untuk instance 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 sebelumnya Klasifikasi Gambar, Deteksi Objek File flatbuffer definisi satu model (.tflite)
XGBoost 1.3 Mendukung 1.3 atau sebelumnya Pohon Keputusan Satu file XGBoost model (.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)