python - 如何在 amazon sagemaker 上运行自己的 python 代码

Question

我有一个使用 keras 和 tensorflow 后端的 python 代码。由于内存空间不足，我的系统不支持训练此模型。我想使用 Amazon sagemaker。

然而，我找到的所有教程都是关于在 docker 容器中部署模型的。我的模型没有经过训练，我想在 Amazon Sagemaker 上对其进行训练。

有没有办法做到这一点？

编辑：我也可以制作我的 python 代码脚本并在 AWS sagemaker 上运行吗？

标签： pythonamazon-sagemaker

Answer 1

SageMaker 为用户提供了引入他们的自定义训练脚本并使用 SageMaker 上的脚本训练他们的算法的能力，该脚本使用 TensorFlow、MXNet、PyTorch 等框架的预构建容器之一。

请看一下https://github.com/aws/amazon-sagemaker-examples/blob/master/frameworks/tensorflow/get_started_mnist_train.ipynb

它介绍了如何使用 TensorFlow 引入训练脚本并使用 SageMaker 对其进行训练。

存储库中还有其他几个示例，它们将帮助您回答您在 SageMaker 之旅中可能遇到的其他问题。