SageMaker Pipelines：自带模型

Question

我正在阅读 SageMaker 管道文档，发现“框架”可以这样检索

image_uri = sagemaker.image_uris.retrieve(
    framework="xgboost",
    region=region,
    version="1.0-1",
    py_version="py3",
    instance_type="ml.m5.xlarge"
)

不过，我很好奇自带模型的流程是怎样的？例如，假设我想使用 JAX、PyMC3 或其他一些不像 xgboost 那样“开箱即用”的框架。

我想象 SageMaker 提供了一些基类，我可以从它继承并将包作为 iniT 参数传递（或者以其他方式向 SageMaker 注册模型。）

将外部模型引入 SageMaker 以与 SageMaker 管道集成的公认做法是什么？

注意：目的是将此类模型/框架集成到管道中，以执行预处理训练等步骤。这与托管在训练管道之外训练的模型形成对比。

Answer 1

有关自带模型，请参阅此链接：https://github.com/aws/amazon-sagemaker-examples/blob/main/advanced_functionity/scikit_bring_your_own/scikit_bring_your_own.ipynb

SageMaker Pipelines 与您使用的模型类型无关。您共享的代码片段会自动获取图像 URI。创建 SageMaker Pipeline 步骤时，您可以使用自己的 ECR 映像。