发明公开
- 专利标题: 深度神经网络模型并行模式选择方法
-
申请号: CN201910897718.8申请日: 2019-09-23
-
公开(公告)号: CN112541584A公开(公告)日: 2021-03-23
- 发明人: 刘鑫 , 刘沙 , 彭超 , 朱传家 , 陈德训 , 黄则强 , 陆旭峰 , 裴阳
- 申请人: 无锡江南计算技术研究所
- 申请人地址: 江苏省无锡市滨湖区山水东路699号
- 专利权人: 无锡江南计算技术研究所
- 当前专利权人: 无锡江南计算技术研究所
- 当前专利权人地址: 江苏省无锡市滨湖区山水东路699号
- 代理机构: 苏州创元专利商标事务所有限公司
- 代理商 王健
- 主分类号: G06N3/08
- IPC分类号: G06N3/08
摘要:
本发明公开一种深度神经网络模型并行模式选择方法,包括以下步骤:S1、测算整个神经网络模型的总数据量;S2、判断S1中获得的神经网络模型的总数据量是否超过进行训练的单个计算节点的可用内存总量,如果不超过,执行S3,如果超过,执行S4;S3、选择数据并行模式;S4、将神经网络模型的网络层进行切分,根据切分的结果,得到神经网络模型所需要分布的计算节点数量,如果输入参数中的计算节点数量不足模型切分所需节点数量两倍以上,执行S5,否则执行S6;S5、选择模型并行模式;S6、选择包括数据并行和模型并行的混合并行模式。本发明通过对模型参数、超参数和数据量的信息采集和分析,实现分布式扩展并行模式的自动选择,并且保证较高的并行性能。
公开/授权文献
- CN112541584B 深度神经网络模型并行模式选择方法 公开/授权日:2022-10-04