Блог

Главная - Вопросы по программированию - SageMaker object2vec — как загрузить артефакт модели и использовать локально?

SageMaker object2vec — как загрузить артефакт модели и использовать локально?

#amazon-web-services #nlp #amazon-sagemaker #embedding

Вопрос:

Я обучил модель object2vec, используя обучающее изображение 174872318107.dkr.ecr.us-west-2.amazonaws.com/object2vec:1 и скачал model.tar.gz файл на мой локальный компьютер. Оттуда я попытался

 import mxnet as mx

tf = tarfile.open('tmp/model.tar.gz')
tf.extractall()
mx.ndarray.load("model_algo-1")

но я получаю следующую ошибку

 MXNetError: Traceback (most recent call last):
  File "../src/ndarray/ndarray.cc", line 1859
MXNetError: Check failed: header == kMXAPINDArrayListMagic: Invalid NDArray file format

Что я должен делать, чтобы иметь возможность загружать и использовать модель локально (для вывода и т. Д.)? Есть ли где-нибудь, где я могу проверить код, который находится на обучающем изображении?

Метки: