全部产品
Search
文档中心

人工智能平台 PAI:使用自定义组件

更新时间:May 25, 2026

您可以根据具体算法使用场景自定义算法组件,自定义组件创建成功后即可在Designer中使用该组件进行模型训练。本文以创建PyTorch类型的自定义组件,代码目录为OSS Bucket为例,为您介绍如何在Designer中使用自定义组件。

使用限制

仅支持在Designer中使用通用训练资源公共资源组运行自定义组件。

前提条件

已将执行代码上传到了OSS Bucket路径,具体操作请参见上传文件。本示例的执行代码示例为main.pyrequirements.txt

操作步骤

  1. 创建Pytorch类型的自定义组件。

    进入新建组件页面,创建Pytorch类型的自定义组件,详细操作步骤与更多参数配置说明,请参见创建自定义组件,本示例的核心配置要点如下。

    • 执行配置

      参数

      描述

      任务类型

      选择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"

    • 管道及参数配置各参数约束条件配置为:param1范围[1,10]param2范围(0,1)param3枚举test1/test2/test3param5枚举1/2/4/8

      参数

      描述

      输入

      单击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。

      • 约束条件配置:单击默认值后的约束,为每个参数配置约束条件。

      • 是否必填分别配置为:是、是、是、否、否。

    • 训练约束

      参数

      描述

      开启训练约束

      打开开启训练约束开关。

      机器类型

      选择CPU和GPU。

      支持多机

      选择支持

      支持多卡

      选择不支持

  2. 新建空白工作流并双击工作流名称进入工作流页面,具体操作,请参见创建工作流:空白工作流

  3. 本示例的组件名称为component,您可以在左侧组件列表直接将其拖入右侧画布中,即可使用自定义组件,其使用方式和Designer内置算法组件一致。将左侧自定义组件目录中的component(1.0)拖拽至中央画布,右侧将自动展示该组件的参数设置面板,可查看已配置的各参数及其可视化控件。

    当自定义组件创建成功后,Designer中会在算法树中展示当前工作空间中的自定义组件。

    • 自定义组件的输入、输出通道会被转换为Designer组件的输入和输出。拖入画布后,工作流中该组件上方连接训练数据测试数据两个输入节点,下方输出模型数据算法checkpoints两个节点。

    • 自定义组件的参数会转换为Designer组件的可视化控件。自定义组件的参数配置会被转换为 Designer 组件右侧参数设置面板中的对应控件。数值型参数(如 param1,范围[1,10],默认值6)显示为带范围约束的输入框;浮点型参数(如 param2,范围(0,1),默认值0.3)同理;枚举型参数(如 param3param5)显示为下拉框;布尔型参数(如 param4)显示为复选框。面板下方还包含 model 模型数据和 checkpoints 算法checkpoints 的可选 OSS 路径输入框,以及 OSS RAM 角色授权状态。

    • 每个输出通道会对应一个OSS目录选择控件,让您可以选择对应的输出目录。例如,model(模型数据)和checkpoints(算法checkpoints)输出通道分别对应独立的OSS路径选择输入框,位于组件右侧参数设置面板的输出配置区域。

    • 自定义组件训练约束配置转换为Designer组件执行调优配置。在画布中选中自定义组件节点后,单击右侧属性面板的执行调优 Tab,可配置资源组(如公共资源组)、机器实例类型(CPU 或 GPU)、机器数量最大运行时长(秒)等执行参数。

后续,如果自定义组件不满足您的使用场景,您还可以在组件列表页面更新自定义组件相关配置或新增组件版本,详情请参见管理自定义组件