如何保存在 Amazon SageMaker 中本地训练的模型？

Question

我正在尝试在 SageMaker 中使用本地培训工作。

遵循此 AWS 笔记本 ( https://github.com/awslabs/amazon-sagemaker-examples/blob/master/sagemaker-python-sdk/mxnet_gluon_mnist/mxnet_mnist_with_gluon_local_mode.ipynb ) 我能够在本地进行训练和预测。

有什么方法可以在本地训练并将训练好的模型保存在 Amazon SageMaker 训练作业部分？否则，如何正确保存我使用本地模式训练的训练模型？

Answer 1

无法让本地模式训练作业出现在 AWS 控制台中。本地模式的目的是在使用 SageMaker 训练模型之前实现更快的迭代/调试。

您可以从本地模型工件创建 SageMaker 模型。将模型工件压缩到文件中.tar.gz，将该文件上传到 S3，然后创建模型（使用 SDK 或在控制台中）。

文档：

• https://sagemaker.readthedocs.io/en/stable/overview.html#using-models-trained-outside-of-amazon-sagemaker
• https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateModel.html