【发布时间】:2018-08-11 05:06:36
【问题描述】:
我使用内置的线性学习器算法训练了一个 AWS SageMaker 模型。我可以从 S3 下载经过训练的模型工件 (model.tar.gz)。
如何在独立于 AWS 的本地环境中部署模型,以便在没有互联网访问的情况下进行预测推理调用?
【问题讨论】:
标签: amazon-sagemaker
我使用内置的线性学习器算法训练了一个 AWS SageMaker 模型。我可以从 S3 下载经过训练的模型工件 (model.tar.gz)。
如何在独立于 AWS 的本地环境中部署模型,以便在没有互联网访问的情况下进行预测推理调用?
【问题讨论】:
标签: amazon-sagemaker
Matx,内置算法没有本地模式。但是,您可以以编程方式加载具有模型权重的 mxnet 模块并使用它来进行预测。代码示例请查看https://forums.aws.amazon.com/thread.jspa?messageID=827236󉽤。
【讨论】:
mxnet.module.Module 类。它可以从 SageMaker Linear Learner 保存的文件中加载模型。另见mxnet.apache.org/api/python/module/…
label_names=["out_label"]作为最后一个参数传递给Module.load。