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