跳到主要内容

创建训练任务详解

这一页把训练创建对话框掰开揉碎:三种打开方式、五个分区里的每个字段都列清楚。快速入门里的创建训练任务是端到端跑通版,这里当字段手册查。

三个创建入口

三个创建入口
  • 项目训练菜单(推荐):项目工作区左侧「训练」菜单,点右上角「训练新模型」。表单全部留空,从零配置。
  • 数据集版本行:数据集详情的「版本」Tab 里,任意就绪版本行点「训练」。系统自动把这一版预填进数据集版本字段,省去下拉选择。
  • 模型版本行(迁移学习):模型详情的版本行点「用此数据训练」。预填该版本训练时的超参数快照(模型架构、轮数、批次、学习率、图像尺寸等),方便复现或微调。

三个入口最终都进同一个对话框,区别只在预填多少。下面以推荐的「项目训练菜单」入口为例走一遍。

操作步骤

  1. 进入项目训练菜单

    打开项目工作区,点左侧导航的「训练」。页面标题「训练任务」,右上角能看到「训练新模型」按钮。

  2. 打开创建对话框

    点「训练新模型」,弹出对话框,标题「训练新模型」,副标题「配置训练参数,开始模型训练」。

  3. 填写基本信息

    在「基本信息」区选数据集版本(必填),任务名可留空走自动生成。

  4. 配置模型

    在「模型配置」区选模型架构;想迁移学习就从「预训练模型」挑一个内置权重。

  5. 设置训练参数

    在「训练参数」区填训练轮数、批次大小、学习率三个核心数字。

  6. 选择设备

    在「硬件配置」区按环境选 GPU、CPU 或 NPU,必要时展开「高级设置」调整图像尺寸。

  7. 点击开始训练

    确认无误后点底部「开始训练」。任务进入「等待中」,再点 ▶ 才会真正开跑。

训练创建对话框
图1:训练创建对话框的五个分区

对话框字段

对话框按五个分区组织:基本信息、模型配置、训练参数、硬件配置、高级设置。

基本信息区

字段类型必填默认值说明
name文本可选自动生成任务名称。留空时系统按「数据集名-任务标签-日期-序号」自动生成,规则见下方提示。
datasetVersionId下拉必填数据集版本。仅显示就绪(ready)且未归档的版本,选项格式为「版本号 · 图片数」。选中后下方会展示类别数与训练集、验证集、测试集样本量。

模型配置区

字段类型必填默认值说明
modelType下拉必填yolov8n模型架构。按项目任务类型自动过滤,例如 YOLOv8 Nano、YOLO11 Small。
modelId下拉可选预训练模型。选「无」表示从头训练;选择内置权重后会自动把 modelType 同步成该权重对应的架构。

训练参数区

字段类型必填默认值说明
epochs数字必填100训练轮数,整个数据集迭代的遍数。建议 100 到 300。
batchSize数字必填16批次大小,每次前向传播处理的图片数。GPU 显存不够时调小。
learningRate数字必填0.01学习率,优化器步长。常用范围 0.001 到 0.01。

硬件配置区

字段类型必填默认值说明
deviceType下拉必填cpu设备类型:GPU、CPU、华为昇腾 NPU。当前环境不支持的选项会标注「(不可用)」并禁用。
设备子选项随类型变化

选不同设备时,下方会冒出对应的附加配置:GPU 有「自动选择 GPU」开关和显存上限;CPU 有 CPU 核心数;昇腾 NPU 有 NPU 内存上限。详细说明见快速入门里的创建训练任务

高级设置区

点对话框底部的「高级设置」条展开。

字段类型必填默认值说明
rectMode开关可选关闭矩形训练模式,适合非正方形图像。开启后图像尺寸改为分别设置宽和高。
imageSize下拉可选640图像尺寸(正方形边长)。可选 320 / 416 / 512 / 640 / 800 / 1024 / 1280。仅矩形模式关闭时显示。
imageWidth数字可选640矩形模式下图像缩放的宽度,步进为 8。仅矩形模式开启时显示。
imageHeight数字可选480矩形模式下图像缩放的高度,步进为 8。仅矩形模式开启时显示。
高级设置展开
图2:高级设置展开(矩形模式与图像尺寸)

几个关键规则

任务名自动生成规则

name 留空时系统自动生成,格式是:

{数据集名}-{任务标签}-{月日}-{当日序号}

  • 任务标签:Det(检测)、Seg(分割)、Pose(姿态)、OBB(旋转框)
  • 月日为两位月加两位日,6 月 17 日写成 0617
  • 当日序号三位补零,从 001 递增

举例:数据集叫 工业缺陷,6 月 17 日当天第一次训检测任务,自动生成 工业缺陷-Det-0617-001

任务类型自动推断

任务类型从项目类型自动推断,对话框里没有手动选择的入口。检测项目推断为 detect、分割推断为 segment、姿态推断为 pose、旋转框推断为 obb。想改任务类型,得在项目层面调整(项目一旦有数据就会锁定类型)。

迁移学习入口的预填

从模型版本行进入时,系统会预填该版本训练时的超参数快照,包括 modelType、epochs、batchSize、learningRate、imageSize / imageWidth / imageHeight、rectMode、deviceType。预填只是个方便起点,所有字段照样能改。

接下来

任务创建好之后怎么看进度、怎么读指标曲线、怎么把训练产物入库,看快速入门里的查看训练进度与指标。想搞懂为什么必须绑定数据集版本,看下一页数据集版本与训练