模型训练
平台支持基于用户选择的数据集进行训练并生成小模型。您可以创建并配置模型训练任务,查看任务进度和模型测试结果,支持增删改查操作。训练成功的模型会存储在【模型管理】-【模型池】中,可发布SDK。
1. 训练任务管理
点击「AI技能」-「模型训练」进入【模型训练】页,默认展示用户新增的训练任务列表,展示信息包括任务名称/任务ID、模型类别、适用型号、任务描述、创建时间和操作(详情、删除)。
- 任务列表支持按照任务名称/ID进行关键字模糊搜索;
- 支持按照模型类别和适用产品进行条件筛选。
1.1 新增训练任务
用户可以在【模型训练】中新增训练任务,后续可在该任务下配置不同版本的训练及增量训练。
【操作步骤】
- 点击「AI技能」-「模型训练」进入【模型训练】页;
- 点击「新增训练任务」,出现弹窗,需填写任务名称、模型类别、适用设备和任务描述;
训练完成后模型名称与该任务名称一致,请谨慎填写。
- 填写完各必填项后,点击「创建」成功创建训练任务,返回任务列表新增一条。
1.2 查看训练详情
用户点击具体任务「详情」按钮,进入训练任务详情页面,用户可以查看当前任务的基本信息和版本列表。
- 任务详情:展示任务名称、任务ID、模型类别、适用设备、适用信号、创建时间和任务描述,点击「编辑」按钮可编辑任务描述;
- 版本列表:展示版本、前序版本、训练状态、训练集、测试集、模型信息、添加时间、训练描述、操作(详情、增量训练和删除);详细字段说明见 2. 训练版本管理
1.3 删除训练任务
在【模型训练】中,点击单个任务操作栏「删除」将删除当前训练任务的所有版本,删除后不可恢复。若有任务版本正在训练则不可删除,会提示您删除失败。
2. 训练版本管理
2.1 添加训练
用户可以在单个训练任务下添加多个训练,配置数据集、训练集、准出标准等信息即可快速训练。
【操作步骤】
- 在【模型训练】页,点击具体任务「详情」按钮,进入训练任务详情页面;
- 点击「添加训练」进入添加页,需进行训练配置和准出标准;
- 训练版本由系统自动生成,无需填写;
- 配置测试集和训练集,支持多选数据集,可展开查看该数据集下的所有的数据组,勾选数据集则数据集下的所有数据全部勾选,没有数量限制;支持筛选数据集类别和模糊搜索名称;
- 配置准出标准,系统给出推荐的准确率和召回率阈值,可自行调整;可选填训练描述,字符限制在50以内;
- 完成必填项后,点击「开始训练」可前往训练版本列表查看进度
【高级配置】
在添加训练时,用户可以选择高级配置,支持选择输入大小,系统相应配置batch数量、通道数量、图像高度和图像宽度等信息;输入迭代次数,限制1-100次;
2.2 增量训练
已经训练成功的版本,用户可以添加增量训练,对模型进行迭代,操作步骤与「添加训练」基本一致。
【操作步骤】
- 在【模型训练】页,点击具体任务「详情」按钮,进入训练任务详情页面;
- 点击单个版本操作栏中「增量训练」进入添加页,需进行训练配置和准出标准;
- 训练版本由系统自动生成,无需填写;
- 选择前序版本,默认选择已经产出的模型最新版本
- 配置测试集和训练集,支持多选数据集;
- 配置准出标准,系统给出推荐的准确率和召回率阈值,可自行调整;可选填训练描述,字符限制在50以内;
- 完成必填项后,点击「开始训练」可前往训练版本列表查看进度。
【高级配置】
在添加训练时,用户可以选择高级,配置添加增量训练时,仅支持用户修改迭代次数,无法修改输入大小;
2.3 版本列表字段说明
在单个任务下添加训练,版本列表会新增一条,列表展示包括版本、前序版本、训练状态、训练集、测试集、模型信息、添加时间、训练描述、操作(详情、增量训练和删除)。
字段名 | 描述 | 备注 |
---|---|---|
版本 | 当前训练的版本号 | -- |
前序版本 | 当前训练若为增量训练,展示训练版本号 | -- |
训练状态 | 当前训练状态,训练中、训练成功、训练失败 | |
训练集 | 训练集名称及版本号 | -- |
测试集 | 测试集名称及版本号 | -- |
模型信息 | 训练成功的模型名称及版本 | 点击模型跳转到对应模型⻚面 |
添加时间 | 当前版本的创建时间 | 默认按照由近及远排序,可进行升降序 |
训练描述 | 用户自定义的训练描述 |
2.4 查看训练详情
在训练任务详情页,点击单个训练版本操作栏「详情」右侧出现训练版本详情,包括基本信息和模型信息。
- 基本信息:训练版本、前序版本、训练集、测试集、添加时间、训练描述、添加时间、输入大小和迭代次数;点击「编辑基本信息」可编辑训练描述。
- 模型信息:模型耗时(在设备上跑通时间)、准确率、召回率和F1分数。
2.5 删除训练版本
点击单个训练版本操作栏「删除」可删除该版本的训练任务,删除后记录不可恢复,但生成的模型文件不会被删除。
任务正在训练中则不可删除