借助自定义训练作业,您可以在 Vertex AI 中运行自定义机器学习 (ML) 训练代码。
CustomTrainingJobOp
CustomTrainingJobOp
组件公开了 CustomJob
资源的全部功能,以允许使用 ContainerSpec
或 PythonPackageSpec
实例进行单一和分布式训练。
create_custom_training_job_from_component
函数
create_custom_training_job_from_component
实用程序将给定容器或 Python 组件转换为在 Vertex AI 中运行自定义作业的组件。这样可以简化自定义训练作业的创建。所提供组件的所有输入和输出都将复制到构造的训练作业运算符中。
请注意,此实用程序会构建 ClusterSpec
,其中主实例和所有工作器使用相同的规范,这意味着所有磁盘和机器规范相关参数将应用于所有副本。这适用于一些使用场景,例如,您使用 MultiWorkerMirroredStrategy
或 MirroredStrategy
进行训练。
此组件不支持 CustomJob
Python 软件包训练,也不支持使用不同工作器池规范的分布式训练。
API 参考文档
- 如需查看组件参考文档,请参阅 CustomJob 组件的 Google Cloud 流水线组件 SDK 参考文档。
- 如需了解 Vertex AI API 参考文档,请参阅
CustomJob
资源页面。
版本历史记录和版本说明
如需详细了解 Google Cloud 流水线组件 SDK 的版本历史记录和更改,请参阅 Google Cloud 流水线组件 SDK 版本说明。
技术支持联系人
如果您有任何疑问,请联系 kubeflow-pipelines-components@google.com。