有效

一种面向并行训练的神经网络模型划分方法、系统、设备及介质

卢凯、李东升、赖志权、郝彦琦、乔林波、贾孟涵
中国人民解放军国防科技大学

摘要

本申请公开了一种面向并行训练的神经网络模型划分方法、系统、设备及介质,通过获得待训练的神经网络模型、训练数据和集群配置;根据所述集群配置中的链路种类,确定通信组;基于所述通信组和所述训练数据,对所述待训练的神经网络模型进行训练,直至得到训练完成的神经网络模型。对于分片数据并行和数据并行各自通信过程,模型不同的参数可以分别使用计算节点内与节点间的对应链路同时进行通信,将原本空闲链路利用起来,减少了原本链路上的通信量,进而减少总的通信开销,提高模型训练速度。