SageMaker создает PyTorchModel без развертывания

Если я создам экземпляр объекта SageMaker PyTorchModel следующим образом:

from sagemaker.pytorch import PyTorchModel

model = PyTorchModel(name=name_from_base('model-name'),
                     model_data=model_data,
                     role=role,
                     framework_version='1.0.0',
                     entry_point='serve.py',
                     source_dir='src',
                     sagemaker_session=sagemaker_session,
                     predictor_cls=ImagePredictor)

#model.create_without_deploying??

Есть ли способ создать эту модель с помощью sagemaker python SDK, чтобы модель отображалась в консоли SageMaker, но без фактического развертывания ее на конечной точке?


python amazon-sagemaker
person Austin    schedule 07.06.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Я не думаю, что это возможно сделать с помощью высокоуровневого SDK SageMaker Pyhton. Однако у вас должна быть возможность сделать это, вызвав API CreateModel с помощью низкоуровневого boto3 https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html.#SageMaker.Client.create_model. Для справки ниже приведен пример фрагмента кода о том, как это сделать.

%%time
import boto3
import time

sage = boto3.Session().client(service_name='sagemaker')

image_uri = '520713654638.dkr.ecr.us-east-1.amazonaws.com/sagemaker-pytorch:1.0.0-cpu-py3'
model_data ='s3://<bucket>/<prefix>/output/model.tar.gz'
source = 's3://<bucket>/<prefix>/sourcedir.tar.gz'
role = 'arn:aws:iam::xxxxxxxx:role/service-role/AmazonSageMaker-ExecutionRole-xxxxxx'

timestamp = time.strftime('-%Y-%m-%d-%H-%M-%S', time.gmtime())
model_name = 'my-pytorch-model' + timestamp

response = sage.create_model(
    ModelName=model_name,
    PrimaryContainer={
        'Image': image_uri,
        'ModelDataUrl': model_data,
        'Environment': { 'SAGEMAKER_CONTAINER_LOG_LEVEL':'20', 'SAGEMAKER_ENABLE_CLOUDWATCH_METRICS': 'False', 
                   'SAGEMAKER_PROGRAM': 'generate.py','SAGEMAKER_REGION': 'us-east-1','SAGEMAKER_SUBMIT_DIRECTORY': source}
         },
         ExecutionRoleArn=role
}
print(response)

Если вы не получите сообщения об ошибке, модель отобразится в консоли SageMaker.

person Christian-kam    schedule 27.06.2019