摘要
本发明提供一种模型训练方法、装置、存储介质及电子设备,该方法包括:获取第一模型,以及获取训练数据集;调用第一模型,分别确定训练数据集中每个训练数据的模型运行结果;并基于每个训练数据的模型运行结果,计算各个模型参数在当前迭代下的一阶梯度;确定各个模型参数在上一次迭代下的一阶梯度,并基于各个模型参数在上一次迭代下的一阶梯度和各个模型参数在当前迭代下的一阶梯度,分别计算各个模型参数在当前迭代下的二阶梯度;基于各个模型参数在当前迭代下的一阶梯度和二阶梯度,分别计算各个模型参数在当前迭代下的模型参数值,以得到第二模型,从而基于第二模型,确定目标模型。本发明实施例可提高模型训练的效率。
技术关键词
参数
阶梯
模型训练方法
数据
处理单元
模型训练装置
矫正
电子设备
计算机
处理器
指令
可读存储介质
程序
指数
存储器
标签