自訂訓練工作可讓您在 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 參考資料
- 如需元件參考資料,請參閱「Google Cloud CustomJob 元件的 SDK 參考資料」。
- 如需 Vertex AI API 參考資料,請參閱
CustomJob
資源頁面。
版本記錄和版本資訊
如要進一步瞭解版本記錄和 Google Cloud Pipeline Components SDK 的變更,請參閱 Google Cloud Pipeline Components SDK 版本資訊。
技術支援聯絡人
如有任何問題,請洽詢 kubeflow-pipelines-components@google.com。