本发明公开了目标检测模型学习率的设置方法和装置,涉及计算机技术领域。该方法的一具体实施方式包括设置训练集总图片数量和每轮迭代训练的图片数量,以计算步长;初始化学习率,以在训练集上进行目标检测模型的训练,且每迭代一个步长将学习率增大预设数值,同时计算一次验证集的平均精度均值,进而获得最小学习率和最大学习率;再以预设数量个步长为一个周期循环训练目标检测模型,直到完成预设的最大迭代次数;其中,基于最小学习率和最大学习率,根据步长内当前迭代的次数,计算每次迭代的学习率。从而,本发明的实施方式能够解决现有技术训练目标检测模型的学习率设置效率低的问题。
声明:
“目标检测模型学习率的设置方法和装置” 该技术专利(论文)所有权利归属于技术(论文)所有人。仅供学习研究,如用于商业用途,请联系该技术所有人。
我是此专利(论文)的发明人(作者)