资讯 文档
技术能力
语音技术
文字识别
人脸与人体
图像技术
语言与知识
视频技术

模型训练

平台支持基于用户选择的数据集进行训练并生成小模型。您可以创建并配置模型训练任务,查看任务进度和模型测试结果,支持增删改查操作。训练成功的模型会存储在【模型管理】-【模型池】中,可发布SDK。

image.png

1. 训练任务管理

点击「AI技能」-「模型训练」进入【模型训练】页,默认展示用户新增的训练任务列表,展示信息包括任务名称/任务ID、模型类别、适用型号、任务描述、创建时间和操作(详情、删除)

image.png

  • 任务列表支持按照任务名称/ID进行关键字模糊搜索;
  • 支持按照模型类别和适用产品进行条件筛选。

1.1 新增训练任务

用户可以在【模型训练】中新增训练任务,后续可在该任务下配置不同版本的训练及增量训练。

【操作步骤】

  1. 点击「AI技能」-「模型训练」进入【模型训练】页;
  2. 点击「新增训练任务」,出现弹窗,需填写任务名称、模型类别、适用设备和任务描述;

image.png

image.png

训练完成后模型名称与该任务名称一致,请谨慎填写。

  1. 填写完各必填项后,点击「创建」成功创建训练任务,返回任务列表新增一条。

1.2 查看训练详情

用户点击具体任务「详情」按钮,进入训练任务详情页面,用户可以查看当前任务的基本信息和版本列表。

  • 任务详情:展示任务名称、任务ID、模型类别、适用设备、适用信号、创建时间和任务描述,点击「编辑」按钮可编辑任务描述;
  • 版本列表:展示版本、前序版本、训练状态、训练集、测试集、模型信息、添加时间、训练描述、操作(详情、增量训练和删除);详细字段说明见 2. 训练版本管理

image.png

1.3 删除训练任务

在【模型训练】中,点击单个任务操作栏「删除」将删除当前训练任务的所有版本,删除后不可恢复。若有任务版本正在训练则不可删除,会提示您删除失败。

image.png

2. 训练版本管理

2.1 添加训练

用户可以在单个训练任务下添加多个训练,配置数据集、训练集、准出标准等信息即可快速训练。

【操作步骤】

  1. 在【模型训练】页,点击具体任务「详情」按钮,进入训练任务详情页面;
  2. 点击「添加训练」进入添加页,需进行训练配置和准出标准;

image.png

  1. 训练版本由系统自动生成,无需填写;

image.png

  1. 配置测试集和训练集,支持多选数据集,可展开查看该数据集下的所有的数据组,勾选数据集则数据集下的所有数据全部勾选,没有数量限制;支持筛选数据集类别和模糊搜索名称;

image.png

image.png

  1. 配置准出标准,系统给出推荐的准确率和召回率阈值,可自行调整;可选填训练描述,字符限制在50以内;

image.png

  1. 完成必填项后,点击「开始训练」可前往训练版本列表查看进度

【高级配置】

在添加训练时,用户可以选择高级配置,支持选择输入大小,系统相应配置batch数量、通道数量、图像高度和图像宽度等信息;输入迭代次数,限制1-100次;

image.png

2.2 增量训练

已经训练成功的版本,用户可以添加增量训练,对模型进行迭代,操作步骤与「添加训练」基本一致。

【操作步骤】

  1. 在【模型训练】页,点击具体任务「详情」按钮,进入训练任务详情页面;
  2. 点击单个版本操作栏中「增量训练」进入添加页,需进行训练配置和准出标准;
  3. 训练版本由系统自动生成,无需填写;
  4. 选择前序版本,默认选择已经产出的模型最新版本

image.png

  1. 配置测试集和训练集,支持多选数据集;

image.png

image.png

  1. 配置准出标准,系统给出推荐的准确率和召回率阈值,可自行调整;可选填训练描述,字符限制在50以内;
  2. 完成必填项后,点击「开始训练」可前往训练版本列表查看进度。

【高级配置】

在添加训练时,用户可以选择高级,配置添加增量训练时,仅支持用户修改迭代次数,无法修改输入大小;

image.png

2.3 版本列表字段说明

在单个任务下添加训练,版本列表会新增一条,列表展示包括版本、前序版本、训练状态、训练集、测试集、模型信息、添加时间、训练描述、操作(详情、增量训练和删除)

字段名 描述 备注
版本 当前训练的版本号 --
前序版本 当前训练若为增量训练,展示训练版本号 --
训练状态 当前训练状态,训练中、训练成功、训练失败
  • 训练中无法删除当前版本训练;
  • 训练成功版本才支持「增量训练」操作
  • 训练失败时会提示失败原因,可按需修改
  • 训练集 训练集名称及版本号 --
    测试集 测试集名称及版本号 --
    模型信息 训练成功的模型名称及版本 点击模型跳转到对应模型⻚面
    添加时间 当前版本的创建时间 默认按照由近及远排序,可进行升降序
    训练描述 用户自定义的训练描述

    2.4 查看训练详情

    在训练任务详情页,点击单个训练版本操作栏「详情」右侧出现训练版本详情,包括基本信息和模型信息。

    • 基本信息:训练版本、前序版本、训练集、测试集、添加时间、训练描述、添加时间、输入大小和迭代次数;点击「编辑基本信息」可编辑训练描述。
    • 模型信息:模型耗时(在设备上跑通时间)、准确率、召回率和F1分数。

    image.png

    image.png

    2.5 删除训练版本

    点击单个训练版本操作栏「删除」可删除该版本的训练任务,删除后记录不可恢复,但生成的模型文件不会被删除。

    image.png

    任务正在训练中则不可删除

    上一篇
    数据管理
    下一篇
    模型管理