一种分布式训练中梯度同步方法及装置
基本信息
| 申请号 | CN201910894224.4 | 申请日 | - |
| 公开（公告）号 | CN110619388A | 公开（公告）日 | 2019-12-27 |
| 申请公布号 | CN110619388A | 申请公布日 | 2019-12-27 |
| 分类号 | G06N3/04（2006.01）; G06N3/08（2006.01） | 分类 | 计算；推算；计数; |
| 发明人 | 李鑫; 王洪伟; 李长亮 | 申请（专利权）人 | 成都金山数字娱乐科技有限公司 |
| 代理机构 | 北京智信禾专利代理有限公司 | 代理人 | 北京金山数字娱乐科技有限公司; 成都金山数字娱乐科技有限公司 |
| 地址 | 100085 北京市海淀区小营西路33号金山软件大厦2层西区 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本申请提供一种分布式训练中梯度同步方法及装置，其中所述分布式训练中梯度同步方法包括：对分布式训练集群中每个训练节点上的训练数据进行分组，获得每个训练节点上的多个子训练数据，其中，分布式训练集群中的训练节点成环形连接；计算所述分布式训练集群的训练节点中每个子训练数据的子训练平均梯度；根据所述子训练平均梯度获得与所述子训练平均梯度对应的子训练累加梯度；将所述子训练累加梯度同步至所述分布式训练集群的每个训练节点。在每个训练节点上将计算不同批量大小训练数据的平均梯度，可以保证梯度的震荡范围较小，能更准确的确定梯度下降的方向，加快模型的训练速度，提高模型训练的效率。 |
