- 专利标题: 分布式训练系统、神经网络模型训练方法、设备和介质
-
申请号: CN202310928151.2申请日: 2023-07-27
-
公开(公告)号: CN116644804B公开(公告)日: 2024-01-12
- 发明人: 闫瑞栋 , 郭振华 , 李仁刚 , 邱志勇
- 申请人: 浪潮电子信息产业股份有限公司
- 申请人地址: 山东省济南市高新区浪潮路1036号
- 专利权人: 浪潮电子信息产业股份有限公司
- 当前专利权人: 浪潮电子信息产业股份有限公司
- 当前专利权人地址: 山东省济南市高新区浪潮路1036号
- 代理机构: 北京集佳知识产权代理有限公司
- 代理商 李慧引
- 主分类号: G06N3/098
- IPC分类号: G06N3/098
摘要:
本发明公开了一种分布式训练系统、神经网络模型训练方法、设备和介质,涉及计算机技术领域,该系统包括控制器和多个计算节点;控制器用于获取目标神经网络模型和训练数据集,并向多个计算节点发送对应的节点训练模型和节点训练数据集;目标神经网络模型为初始的神经网络模型去除第i个神经网络层与第i+1个神经网络层之间链接重要程度最低的第一数量个链接、去除神经元重要程度最低的第二数量个神经元以及对应的链接得到神经网络模型;计算节点用于基于接收到的节点训练数据集和接收到的节点训练模型的当前模型参数计算局部信息,并基于全局信息更新节点训练模型的模型参数,实现节点训练模型的迭代训练,得到训练完成的节点训练模型。
公开/授权文献
- CN116644804A 分布式训练系统、神经网络模型训练方法、设备和介质 公开/授权日:2023-08-25