模型训练-影像分析
使用说明
模型训练的目的是使用生成的训练数据进行神经网络模型训练。整体训练过程通过多次迭代(epoch)的方式来得到训练结果更优的网络模型,同时,通过超参数调优(学习率、单步运算量等)可提升模型训练效率和准确度,以得到可用的神经网络模型。
此外,也可以使用少量的样本数据对预训练模型进行微调,使模型更好地适应具体应用场景,同时减少训练时间。
参数说明
参数名 | 默认值 | 参数释义 | 参数类型 |
---|---|---|---|
训练数据路径 | 指定生成的训练数据路径 | String | |
训练模型用途 | 目标检测 | 指定训练数据用途 | Object |
模型算法 | 根据模型用途选择合适的模型算法 | String | |
训练配置文件 | 指定训练配置文件路径 | String | |
训练次数 | 10 | 模型训练的次数(epoch),默认值为 10。随着训练次数的增多,模型拟合度越大,但次数太多可能会出现过拟合的情况,用户可根据需求选择,训练次数与运行花费时间成正比。最多训练300次,超过此数则不具有训练意义。 | Integer |
单步运算量 | 1 | 一次训练中单步运算的图片数量(Batch Size)。在合理范围内,单步运算的数量与内存(显存)占用成正比,与训练时间成反比。 | Integer |
学习率 | 模型参数的更新幅度。若难以确定合适的学习率,可不填写该参数,产品内设了基准学习率和基准单步运算量,将根据用户设置的单步运算量使用合适的学习率。 | Double | |
训练日志路径 | 指定训练日志的保存路径。 | String | |
加载预训练模型 | 勾选可添加预训练模型路径,进行迁移学习 | Boolean | |
预训练模型 | 指定用于迁移学习的预训练模型文件(*.sdm) | String | |
微调方法 | 全量微调 | 对于Segformer和Mask2Former模型,支持全量微调和LoRA微调两种方法;其他模型仅支持全量微调。LoRA微调适用于大参数量模型的高效迁移训练,降低大参数量模型迁移学习资源消耗。 | String |
处理器类型 | GPU | 指定处理器类型 | String |
GPU编号 | 0 | 输入GPU编号。如需多卡推理,可以输入多个GPU编号,如输入“0,1,2”。 | String |
模型存储路径 | 指定模型存储路径 | String | |
模型名 | saved_model | 指定模型名称 | String |
输出结果
模型训练生成的模型文件及日志。