[发明专利]一种分布式训练中参数更新方法及装置有效
申请号: | 201910864527.1 | 申请日: | 2019-09-12 |
公开(公告)号: | CN110633798B | 公开(公告)日: | 2023-04-07 |
发明(设计)人: | 王洪伟;李鑫;李长亮 | 申请(专利权)人: | 北京金山数字娱乐科技有限公司;成都金山数字娱乐科技有限公司 |
主分类号: | G06N3/08 | 分类号: | G06N3/08 |
代理公司: | 北京智信禾专利代理有限公司 11637 | 代理人: | 王治东 |
地址: | 100085 北京市海淀*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请提供一种分布式训练中参数更新方法及装置,其中所述分布式训练中参数更新方法包括:接收分布式训练集群中的每个训练节点传递的梯度;根据每个训练节点传递的梯度计算训练模型的损失值并更新所述训练模型的模型参数;根据所述损失值确定模型训练的训练阶段;根据所述训练阶段对所述模型参数进行压缩,获得压缩模型参数;将所述压缩模型参数同步至所述分布式训练集群的每个训练节点,更新每个训练节点中的模型参数。本申请提供的分布式训练中参数更新方法自适应地减少了参数同步时通信量的大小,即保证了模型参数的同步效率,又保证了训练结果的精度,在进行分布式训练时,提高了模型训练的速度。 | ||
搜索关键词: | 一种 分布式 训练 参数 更新 方法 装置 | ||
【主权项】:
1.一种分布式训练中参数更新方法,其特征在于,包括:/n接收分布式训练集群中的每个训练节点传递的梯度;/n根据每个训练节点传递的梯度计算训练模型的损失值并更新所述训练模型的模型参数;/n根据所述损失值确定模型训练的训练阶段;/n根据所述训练阶段对所述模型参数进行压缩,获得压缩模型参数;/n将所述压缩模型参数同步至所述分布式训练集群的每个训练节点,更新每个训练节点中的模型参数。/n
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京金山数字娱乐科技有限公司;成都金山数字娱乐科技有限公司,未经北京金山数字娱乐科技有限公司;成都金山数字娱乐科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910864527.1/,转载请声明来源钻瓜专利网。