基于智能网卡的深度学习训练网络迭代更新方法

摘要

本发明涉及一种基于智能网卡的深度学习训练网络迭代更新方法，包括将所需要训练的模型与数据分发至训练网络中的训练节点；各训练节点根据自己得到的数据与模型，进行前向和后向运算，得到梯度结果；所述各训练节点得到训练的梯度时，选择合适的时机进行广播；所述各训练节点得到其他节点的梯度，由智能网卡进行累加，汇集结果，进行模型参数更新。本发明提供一种基于智能网卡的深度学习训练网络，采用建立局域网，在局域网内广播的方式进行梯度的传递，以在采用多主机分布式训练时，有效提高迭代的速度。