全部產品
Search
文件中心

Platform For AI:使用自訂群組件

更新時間:Jul 13, 2024

您可以根據具體演算法使用情境自訂演算法組件,自訂群組件建立成功後即可在Designer中使用該組件進行模型訓練。本文以建立PyTorch類型的自訂群組件,代碼目錄為OSS Bucket為例,為您介紹如何在Designer中使用自訂群組件。

使用限制

僅支援在Designer中使用通用訓練資源公用資源群組運行自訂群組件。

前提條件

已將執行代碼上傳到了OSS Bucket路徑,具體操作請參見控制台上傳檔案。本樣本的執行程式碼範例為main.pyrequirements.txt

操作步驟

  1. 建立Pytorch類型的自訂群組件。

    進入新群組件頁面,建立Pytorch類型的自訂群組件,詳細操作步驟與更多參數配置說明,請參見建立自訂群組件,本樣本的核心配置要點如下。

    • 執行配置image..png

      參數

      描述

      任務類型

      選擇Pytorch。

      執行鏡像

      選擇社區鏡像pytorch-training:1.12-cpu-py39-ubuntu20.04

      執行代碼

      配置為代碼入口檔案(main.py、requirements.txt)所在的OSS Bucket路徑。

      執行命令

      配置為:python main.py $PAI_USER_ARGS --train $PAI_INPUT_TRAIN --test $PAI_INPUT_TEST --model $PAI_OUTPUT_MODEL --checkpoints $PAI_OUTPUT_CHECKPOINTS && echo "job finished"

    • 管道及參數配置image..png

      參數

      描述

      輸入

      單擊image..png配置兩個輸入管道:

      • 輸入名稱分別配置為:train、test。

      • 輸入來源:OSS。

      • 是否必填分別配置為:是、否。

      • 描述分別配置為:訓練資料、測試資料。

      輸出

      單擊image..png配置兩個輸出管道:

      • 輸出名稱分別配置為:model、checkpoints。

      • 儲存類型:OSS。

      • 是否必填:是。

      • 描述分別配置為:模型資料、演算法checkpoints。

      參數

      單擊image..png配置五個參數:

      • 參數名稱分別配置為:param1、param2、param3、param4、param5。

      • 參數類型分別配置為:Int、Float、String、Bool、Int。

      • 預設值分別配置為:6、0.3、test1、true、2。

      • 約束條件配置:單擊預設值後的約束,分別參照上圖為每個參數配置約束條件。

      • 是否必填分別配置為:是、是、是、否、否。

    • 訓練約束image..png

      參數

      描述

      開啟訓練約束

      開啟開啟訓練約束開關。

      機器類型

      選擇CPU和GPU。

      支援多機

      選擇支援

      支援多卡

      選擇不支援

  2. 建立空白工作流程並雙擊工作流程名稱進入工作流程頁面,具體操作,請參見建立工作流程:空白工作流程

  3. 本樣本的組件名稱為component,您可以在左側組件列表直接將其拖入右側畫布中,即可使用自訂群組件,其使用方式和Designer內建演算法組件一致。7706a160936ede732315086bf3745c83..png

    當自訂群組件建立成功後,Designer中會在演算法樹中展示當前工作空間中的自訂群組件。

    • 自訂群組件的輸入、輸出通道會被轉換為Designer組件的輸入和輸出。40ea48eb12349496759fb55cb857a050..png

    • 自訂群組件的參數會轉換為Designer組件的可視化控制項。image..png

    • 每個輸出通道會對應一個OSS目錄選擇控制項,讓您可以選擇對應的輸出目錄。image..png

    • 自訂群組件訓練約束配置轉換為Designer組件執行調優配置。image..png

後續,如果自訂群組件不滿足您的使用情境,您還可以在組件列表頁面更新自訂群組件相關配置或新增組件版本,詳情請參見管理自訂群組件