기계 번역으로 제공되는 번역입니다. 제공된 번역과 원본 영어의 내용이 상충하는 경우에는 영어 버전이 우선합니다.
지원되는 프레임워크
Amazon SageMaker Neo는 다음 프레임워크를 지원합니다.
| 프레임워크 | 프레임워크 버전 | 모델 버전 | 모델 | 모델 형식(*.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 이하 지원 | 이미지 분류, 객체 감지 | 모델 정의 플랫 버퍼 파일(.tflite) 한 개 | |
| XGBoost | 1.3 | 1.3 이하 지원 | 의사결정 트리 | 트리의 노드 수가 2^31 미만인 XGBoost 모델 파일 (.model) 1개 | |
| DARKNET | 이미지 분류, 객체 감지(Yolo 모델은 지원되지 않음) | 구성(.cfg) 파일 하나와 가중치(.weight) 파일 하나 |
