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