(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111683315.7 (22)申请日 2021.12.31 (71)申请人 苏州浪潮智能科技有限公司 地址 215000 江苏省苏州市吴中经济开发 区郭巷街道官浦路1号9幢 (72)发明人 辛永欣　 (74)专利代理 机构 北京三聚阳光知识产权代理 有限公司 1 1250 专利代理师 陈刚 (51)Int.Cl. G06K 9/62(2022.01) G06N 3/08(2006.01) (54)发明名称 模型训练方法、 装置、 计算机设备及存储介 质 (57)摘要 本发明实施例涉及一种模型训练方法、 装 置、 计算机设备及存储介质， 所述方法包括： 获取 预设的超参数信息、 基本参数信息以及训练数据 信息； 基于所述超参数信息、 基本参数信息以及 训练数据信息对初始模型进行训练； 在所述初始 模型的第一输出数据达到第一目标条件后， 调整 所述超参数信息， 继续对所述初始模型进行训 练， 直至所述初始模型的第二输出数据达到第二 目标条件， 则确定所述初始模型训练完成， 由本 方法， 采用混合精度的方式， 在将模型训练分成 两个阶段， 不同阶段采用不同的训练精度， 可 以 实现不降低模型训练精度的前提下减少模型训 练占用的资源。 权利要求书2页 说明书8页 附图2页 CN 114511000 A 2022.05.17 CN 114511000 A 1.一种模型训练方法， 其特 征在于， 包括： 获取预设的超参数信息、 基本参数信息以及训练数据信息； 基于所述超参数信息、 基本参数信息以及训练数据信息对初始模型进行训练； 在所述初始模型的第一输出数据达到第一目标条件后， 调整所述超参数信息， 继续对 所述初始模型进行训练， 直至所述初始模型 的第二输出数据达到第二 目标条件， 则确定所 述初始模型训练完成。 2.根据权利要求1所述的方法， 其特征在于， 所述在所述初始模型的第 一输出数据达到 第一目标 数据后， 调整所述超参数信息， 包括： 获取所述初始模型在训练过程中输出的第一输出 数据； 基于所述第一输出 数据， 对比第一预设训练周期内的第一输出 数据的第一差值信息； 若所述第一差值信息小于或等于第一阈值， 则确定调整所述超参数信息 。 3.根据权利要求2所述的方法， 其特 征在于， 所述方法还 包括： 若所述第一差值信息大于第一阈值， 则确定对所述初始模型继续训练。 4.根据权利要求3所述的方法， 其特征在于， 所述初始模型的第 二输出数据达到第 二目 标条件， 则确定所述初始模型训练完成， 包括： 获取所述初始模型在调整所述超参数信息后进行训练过程中输出的第二输出 数据； 基于所述第二输出 数据， 对于第二预设训练周期内的第二输出 数据的第二差值信息； 若所述第二差值信息小于或等于第一阈值， 则确定所述初始模型训练完成。 5.根据权利要求 4所述的方法， 其特 征在于， 所述方法还 包括： 若所述第二差值信息大于第一阈值， 则确定对所述初始模型继续训练。 6.一种模型训练装置， 其特 征在于， 包括： 获取模块， 用于获取 预设的超参数信息、 基本参数信息以及训练数据信息； 训练模块， 用于基于所述超参数信息、 基本参数信息以及训练数据信息对初始模型进 行训练； 所述训练模块， 还用于在所述初始模型的第一输出数据达到第一目标条件后， 调整所 述超参数信息， 继续对所述初始模型进行训练， 直至所述初始模型 的第二输出数据达到第 二目标条件， 则确定所述初始模型训练完成。 7.根据权利要求6所述的模型训练装置， 其特征在于， 所述训练模块， 具体用于获取所 述初始模型在训练过程中输出 的第一输出数据； 基于所述第一输出数据， 对比第一预设训 练周期内的第一输出数据的第一差值信息； 若所述第一差值信息小于或等于第一阈值， 则 确定调整所述超参数信息 。 8.根据权利要求6所述的模型训练装置， 其特征在于， 所述训练模块， 具体用于获取所 述初始模型在调整所述超参数信息后进行训练过程中输出的第二输出数据； 基于所述第二 输出数据， 对于第二预设训练周期内的第二输出数据的第二差值信息； 若所述第二差值信 息小于或等于第一阈值， 则确定所述初始模型训练完成。 9.一种计算机设备， 其特征在于， 包括： 处理器和存储器， 所述处理器用于执行所述存 储器中存 储的模型训练程序， 以实现权利要求1～5中任一项所述的模型训练方法。 10.一种存储介质， 其特征在于， 所述存储介质存储有一个或者多个程序， 所述一个或 者多个程序可被一个或者多个处理器执行， 以实现权利要求 1～5中任一项所述的模型训练权　利　要　求　书 1/2 页 2 CN 114511000 A 2方法。权　利　要　求　书 2/2 页 3 CN 114511000 A 3