AWS sagemaker中指定的代码的入口点如何带来您自己的代码?

时间:2018-02-27 10:14:46

标签: amazon-web-services aws-sdk amazon-sagemaker

我正在试用AWS Sagemaker中的示例笔记本,目前在mxnet mnist示例中演示了自带代码。在实例化估算器实例时传入的入口点参数,仅提及源文件(mnist.py),而不是方法名称或源文件中的任何其他点。

那么aws sagemaker如何找出将训练数据发送到哪个方法?

2 个答案:

答案 0 :(得分:2)

您的python脚本应该实现一些方法,如train,model_fn,transform_fn,input_fn等.SagaMaker会在需要时调用适当的方法。

https://docs.aws.amazon.com/sagemaker/latest/dg/mxnet-training-inference-code-template.html

答案 1 :(得分:0)

这是我找到的答案。对于自带算法的用例,SageMaker将运行一个名为" train"的可执行程序。用于培训和"服务"用于托管。这个example提供了更多细节。或者,您也可以在Dockerfile中指定任何具有在其中定义的train()和serve()函数的ENTRYPOINT。