模型训练
选择训练参数
数据标注完成后,点击 标注 & 训练 页面右上角的 训练模型 按钮。
AI Creator 针对每一类算法都预置了训练参数,能够满足通用场景的检测需求,同时也会根据标注数据和图像大小,自动优化模型输入、epoch 大小等参数。
TIP
训练参数的详细说明在界面上有 tooltip,您可以将鼠标移动到相应参数的 tooltip 图标上,即可查看详细的参数说明。
由于我们的需求是定位出芯片主体,比较简单,我们使用默认参数即可,点击 开始训练 按钮即可开始模型训练。
训练过程
开始训练后,会自动进入训练详情界面,在训练详情界面可以查看训练的实时日志,例如当前训练到了第多少个 epoch,同时也能实时看到 train/valid loss 以及 IoU 等信息。
训练完成
AI Creator 会自动评估并保存多个训练模型,训练完成后,系统会挑选一个数值最优的模型存档为一个版本,并且会自动基于该模型进行一次 模型评估。
下一步
模型训练完成后,我们就可以进行 模型评估,查看该模型的效果。