Invention Grant
- Patent Title: 模型训练方法、装置、电子设备和计算机可读存储介质
-
Application No.: CN202011552010.8Application Date: 2020-12-24
-
Publication No.: CN112631775BPublication Date: 2024-06-18
- Inventor: 陈庆澍 , 欧阳剑 , 王勇 , 赵志彪
- Applicant: 北京百度网讯科技有限公司
- Applicant Address: 北京市海淀区上地十街10号百度大厦2层
- Assignee: 北京百度网讯科技有限公司
- Current Assignee: 北京百度网讯科技有限公司
- Current Assignee Address: 北京市海淀区上地十街10号百度大厦2层
- Agency: 北京市金杜律师事务所
- Agent 赵林琳; 张昊
- Main IPC: G06F9/50
- IPC: G06F9/50 ; G06N3/08 ; G06N3/084 ; G06N3/10

Abstract:
本公开公开了模型训练方法、装置、电子设备和计算机可读存储介质,涉及人工智能领域,尤其涉及深度学习领域和人工智能芯片领域。具体实现方案为:通过一个计算单元对一个训练数据集执行前向计算操作序列以及反向计算操作序列中包括的第一反向计算操作,以基于第一反向计算操作确定第一计算结果和模型的第一梯度;至少基于第一梯度,确定第一同步梯度;基于所述第一计算结果对所述一个训练数据集执行所述反向计算操作序列中包括的第二反向计算操作,以基于所述第二反向计算操作至少确定所述模型的第二梯度;至少基于第二梯度,确定第二同步梯度;以及至少基于第一同步梯度和第二同步梯度更新模型的参数。以此方式,实现了高效准确的并行训练。
Public/Granted literature
- CN112631775A 模型训练方法、装置、电子设备和计算机可读存储介质 Public/Granted day:2021-04-09
Information query