发布于: Jan 21, 2022

Amazon SageMaker Pipelines是一项全托管式服务,允许客户将其模型构建步骤定义为工作流并进行编排,现在允许客户对可并行执行的步骤数量设置并发限制。

SageMaker Pipelines 提供了多种步骤(例如处理、训练、注册模型、回调等)。这些步骤让客户能灵活地在其 ML 模型构建工作流中定义各种实体。以前,SageMaker Pipelines 在成功完成相关步骤时会执行一个步骤;这可能导致并行执行多个步骤安排的情况。并发控制帮助客户控制可以并行执行的管道步骤的数量。通过这项新功能,客户可以选择为单个步骤或整个管道设置精确并发限制。如果达到并发限制,SageMaker Pipelines 将暂停步骤,并在成功完成全部正在执行的步骤后安排被暂停的步骤。通过这种方式,客户可以避免由于资源不足而导致的故障,当同时安排执行需要相同资源的多个步骤时,可能会遇到这种情况。

此功能将在所有提供 Amazon SageMaker 的 AWS 区域推出。要开启此功能,请从 Amazon SageMaker SDK 或 Studio 创建一个新的 SageMaker 管道,并访问我们关于并发控制的文档页面。