快速开始¶
- 根据任务选择现代架构、经典架构或异常检测。类别判断用分类,普通框用目标检测,掩膜用实例分割,带方向的框用旋转框检测。
- 准备对应格式的数据集。经典检测和分类只需选择一个数据集文件夹;现代架构使用各自的标签格式。
- 选择模型、设备和输出相关选项。经典及异常检测使用各自独立环境;可以从多张 NVIDIA 显卡中选择一张,当前 Windows 方案不支持同一经典模型联合多卡训练。
- 先用较小批次和少量轮数检查路径、标注、显存及输出,再开展正式训练。这里的流程检查不用于判断最终精度。
- 启动后到“控制台日志”查看训练图表和原始日志。图表用于看趋势,原始日志用于查看初始化、验证、导出和错误详情。
- 训练成功后查看自动导出的模型。经典训练完成与导出完成是两个阶段;导出失败会保留已保存的训练权重。
- 到“训练历史”打开实验文件夹,或选择模型进入推理验证。异常检测的推理位于异常检测自己的页面。
启动任务时会读取一份参数。随后修改界面控件不会改变已经启动的任务;修改应在下一次任务启动前完成。
关闭软件会弹出确认,默认选择取消。任务运行期间确认退出会终止任务。软件正在删除训练文件时,需要等待清理完成后再退出。
三类运行环境¶
| 架构 | 训练方式 | 设备选择 |
|---|---|---|
| 四种现代架构 | 现代训练引擎 | auto / gpu / cpu;gpu 为第 0 张 |
| 两种经典架构 | 独立经典环境 | CPU 或任意选择的一张 NVIDIA GPU |
| 异常检测 | 独立异常检测环境 | CPU 或任意选择的一张 NVIDIA GPU |
手动选择 GPU 后,不应把失败理解为“软件已经自动改用 CPU”。经典及异常检测 GPU 不可用时会报错,需查看日志并检查环境。