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 Pipeline Components SDK 的變更,請參閱 Google Cloud Pipeline Components SDK 版本資訊

技術支援聯絡人

如有任何問題,請洽詢 kubeflow-pipelines-components@google.com