创建图像搜索模型训练任务
约束
当前图像搜索模型支持预训练,适用于图像标签类训练任务。
创建图像搜索预训练步骤
创建图像搜索模型预训练步骤如下:
- 登录ModelArts Studio大模型开发平台,在“我的空间”(详见工作空间介绍)分页,单击进入所需空间。
- 在左侧导航栏中选择“模型开发 > 模型训练”,单击右上角“创建训练任务”。
- 在“选择模型”页面,选择“模型广场 > 图像搜索 > 模型”,单击“确定”进入“创建训练任务”页面。
- 在“创建训练任务”页面,参考表1完成训练参数设置,参数默认值在创建训练任务的时候会带出。 
    表1 图像搜索模型预训练参数说明 参数分类 训练参数 参数说明 训练配置 选择模型 可以修改如下信息: - 来源:选择“模型广场”。
- 类型:选择“图像搜索 ”,并选择训练所用的基础模型和版本。
 训练类型 选择“预训练”。 高级设置 checkpoints:在模型训练过程中,用于保存模型权重和状态的机制。 - 关闭:关闭后不保存checkpoints,无法基于checkpoints执行续训操作。
- 自动:自动保存训练过程中的所有checkpoints。
- 自定义:根据设置保存指定数量的checkpoints。
 训练参数 epochs 训练轮次,表示完成全部训练数据集训练的次数。每个轮次都会遍历整个数据集一次。 batch_size 表示每个训练步骤中使用的样本数量。较大的批量可以提供更稳定的梯度,但可能会增加计算资源的使用和训练时间。 training_categories 需要训练的标签,每个标签以逗号隔开,不输入时默认训练数据中的标签全部训练。 数据集 选择已发布的数据集进行本次训练。 资源配置 计费模式 选择训练当前任务的计费模式。 训练单元 选择训练模型所需的训练单元。训练单元 = 单实例训练单元 * 实例数。 当前展示的完成本次训练所需要的最低训练单元要求。 单实例训练单元数 选择单实例训练单元数。 实例数 选择实例数。 当实例数大于1,将启动多节点分布式训练。 订阅提醒 订阅提醒 该功能开启后,系统将在任务状态更新时,通过短信或邮件将提醒发送给用户。 发布模型 开启自动发布 开启自动发布后,模型训练完成的最终产物会自动发布为空间资产,以便对模型进行压缩、部署、评测等操作或共享给其他空间。 基本信息 名称 训练任务名称。 描述 训练任务描述。 
- 参数填写完成后,单击“立即创建”。
- 创建好训练任务后,页面将返回“模型训练”页面,可随时查看当前任务的状态。
 
  