模型训练方法、装置、存储介质及电子设备

AITNT
正文
推荐专利
模型训练方法、装置、存储介质及电子设备
申请号:CN202411967383
申请日期:2024-12-30
公开号:CN120069121A
公开日期:2025-05-30
类型:发明专利
摘要
本发明提供一种模型训练方法、装置、存储介质及电子设备,该方法包括:获取第一模型,以及获取训练数据集;调用第一模型,分别确定训练数据集中每个训练数据的模型运行结果;并基于每个训练数据的模型运行结果,计算各个模型参数在当前迭代下的一阶梯度;确定各个模型参数在上一次迭代下的一阶梯度,并基于各个模型参数在上一次迭代下的一阶梯度和各个模型参数在当前迭代下的一阶梯度,分别计算各个模型参数在当前迭代下的二阶梯度;基于各个模型参数在当前迭代下的一阶梯度和二阶梯度,分别计算各个模型参数在当前迭代下的模型参数值,以得到第二模型,从而基于第二模型,确定目标模型。本发明实施例可提高模型训练的效率。
技术关键词
参数 阶梯 模型训练方法 数据 处理单元 模型训练装置 矫正 电子设备 计算机 处理器 指令 可读存储介质 程序 指数 存储器 标签
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号