CustomJob 组件

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

CustomTrainingJobOp

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

create_custom_training_job_from_component 函数

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

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

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

API 参考文档

版本历史记录和版本说明

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

技术支持联系人

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