我正在试用AWS Sagemaker中的示例笔记本,目前在mxnet mnist示例中演示了自带代码。在实例化估算器实例时传入的入口点参数,仅提及源文件(mnist.py),而不是方法名称或源文件中的任何其他点。
那么aws sagemaker如何找出将训练数据发送到哪个方法?
答案 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。