更新时间:2025-09-16 GMT+08:00
分享

创建图像搜索模型训练任务

约束

当前图像搜索模型支持预训练,适用于图像标签类训练任务。

创建图像搜索预训练步骤

创建图像搜索模型预训练步骤如下:

  1. 登录ModelArts Studio大模型开发平台,在“我的空间”(详见工作空间介绍)分页,单击进入所需空间。
  2. 在左侧导航栏中选择“模型开发 > 模型训练”,单击右上角“创建训练任务”。
  3. 在“选择模型”页面,选择“模型广场 > 图像搜索 > 模型”,单击“确定”进入“创建训练任务”页面。
  4. 在“创建训练任务”页面,参考表1完成训练参数设置,参数默认值在创建训练任务的时候会带出。
    表1 图像搜索模型预训练参数说明

    参数分类

    训练参数

    参数说明

    训练配置

    选择模型

    可以修改如下信息:

    • 来源:选择“模型广场”。
    • 类型:选择“图像搜索 ”,并选择训练所用的基础模型和版本。

    训练类型

    选择“预训练”。

    高级设置

    checkpoints:在模型训练过程中,用于保存模型权重和状态的机制。

    • 关闭:关闭后不保存checkpoints,无法基于checkpoints执行续训操作。
    • 自动:自动保存训练过程中的所有checkpoints。
    • 自定义:根据设置保存指定数量的checkpoints。

    训练参数

    epochs

    训练轮次,表示完成全部训练数据集训练的次数。每个轮次都会遍历整个数据集一次。

    batch_size

    表示每个训练步骤中使用的样本数量。较大的批量可以提供更稳定的梯度,但可能会增加计算资源的使用和训练时间。

    training_categories

    需要训练的标签,每个标签以逗号隔开,不输入时默认训练数据中的标签全部训练。

    数据集

    选择已发布的数据集进行本次训练。

    资源配置

    计费模式

    选择训练当前任务的计费模式。

    训练单元

    选择训练模型所需的训练单元。训练单元 = 单实例训练单元 * 实例数。

    当前展示的完成本次训练所需要的最低训练单元要求。

    单实例训练单元数

    选择单实例训练单元数。

    实例数

    选择实例数。

    当实例数大于1,将启动多节点分布式训练。

    订阅提醒

    订阅提醒

    该功能开启后,系统将在任务状态更新时,通过短信或邮件将提醒发送给用户。

    发布模型

    开启自动发布

    开启自动发布后,模型训练完成的最终产物会自动发布为空间资产,以便对模型进行压缩、部署、评测等操作或共享给其他空间。

    基本信息

    名称

    训练任务名称。

    描述

    训练任务描述。

  5. 参数填写完成后,单击“立即创建”。
  6. 创建好训练任务后,页面将返回“模型训练”页面,可随时查看当前任务的状态。

相关文档