CustomJob 组件

借助自定义训练作业,您可以在 Vertex AI 中运行自定义机器学习 (ML) 训练代码。

create_custom_training_job_from_component

create_custom_training_job_from_component 实用程序将给定容器或 Python 组件转换为在 Vertex AI 中运行自定义作业的组件。这样可以简化自定义训练作业的创建。所提供组件的所有输入和输出都将复制到构造的训练作业运算符中。

请注意,此实用程序会构建 ClusterSpec,其中主实例和所有工作器使用相同的规范,这意味着所有磁盘和机器规范相关参数将应用于所有副本。这适用于一些使用场景,例如,您使用 MultiWorkerMirroredStrategyMirroredStrategy 进行训练。

此组件不支持 CustomJob Python 软件包训练,也不支持使用不同工作器池规范的分布式训练。

CustomJobOp

CustomTrainingJobOp 组件公开了 CustomJob 资源的全部功能,以允许通过 ContainerSpecPythonPackageSpec 进行单一和分布式训练。

API 参考文档

版本历史记录和版本说明

如需详细了解 Google Cloud 流水线组件 SDK 的版本历史记录和更改,请参阅 Google Cloud 流水线组件 SDK 版本说明

技术支持联系人

如果您有任何疑问,请联系 kubeflow-pipelines-components@google.com