我正在 AWS Sagemaker 中试用示例笔记本,目前在 mxnet mnist 示例中,该示例演示了您自己的代码。实例化估算器实例时传入的入口点参数仅提及源文件(mnist.py),而不是方法名称或源文件中的任何其他点。
那么 aws sagemaker 如何确定将训练数据发送到哪种方法呢?
我正在 AWS Sagemaker 中试用示例笔记本,目前在 mxnet mnist 示例中,该示例演示了您自己的代码。实例化估算器实例时传入的入口点参数仅提及源文件(mnist.py),而不是方法名称或源文件中的任何其他点。
那么 aws sagemaker 如何确定将训练数据发送到哪种方法呢?
您的 python 脚本应该实现一些方法,例如 train、model_fn、transform_fn、input_fn 等。SagaMaker 会在需要时调用适当的方法。
https://docs.aws.amazon.com/sagemaker/latest/dg/mxnet-training-inference-code-template.html
这是我找到的答案。对于自带算法的用例,SageMaker 将寻求运行一个名为“train”的可执行程序来进行训练,而“serve”来进行托管。这个例子提供了更多的细节。或者,或者,您可以在 Dockerfile 中指定任何 ENTRYPOINT,其中定义了 train() 和 serve() 函数。