一種分布式訓(xùn)練中梯度同步方法及裝置

基本信息

申請(qǐng)?zhí)?/td> CN201910894224.4 申請(qǐng)日 -
公開(kāi)（公告）號(hào) CN110619388A 公開(kāi)（公告）日 2019-12-27
申請(qǐng)公布號(hào) CN110619388A 申請(qǐng)公布日 2019-12-27
分類(lèi)號(hào) G06N3/04（2006.01）; G06N3/08（2006.01） 分類(lèi) 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 李鑫; 王洪偉; 李長(zhǎng)亮 申請(qǐng)（專(zhuān)利權(quán)）人 成都金山數(shù)字娛樂(lè)科技有限公司
代理機(jī)構(gòu) 北京智信禾專(zhuān)利代理有限公司 代理人 北京金山數(shù)字娛樂(lè)科技有限公司; 成都金山數(shù)字娛樂(lè)科技有限公司
地址 100085 北京市海淀區(qū)小營(yíng)西路33號(hào)金山軟件大廈2層西區(qū)
法律狀態(tài) -

摘要

摘要 本申請(qǐng)?zhí)峁┮环N分布式訓(xùn)練中梯度同步方法及裝置，其中所述分布式訓(xùn)練中梯度同步方法包括：對(duì)分布式訓(xùn)練集群中每個(gè)訓(xùn)練節(jié)點(diǎn)上的訓(xùn)練數(shù)據(jù)進(jìn)行分組，獲得每個(gè)訓(xùn)練節(jié)點(diǎn)上的多個(gè)子訓(xùn)練數(shù)據(jù)，其中，分布式訓(xùn)練集群中的訓(xùn)練節(jié)點(diǎn)成環(huán)形連接；計(jì)算所述分布式訓(xùn)練集群的訓(xùn)練節(jié)點(diǎn)中每個(gè)子訓(xùn)練數(shù)據(jù)的子訓(xùn)練平均梯度；根據(jù)所述子訓(xùn)練平均梯度獲得與所述子訓(xùn)練平均梯度對(duì)應(yīng)的子訓(xùn)練累加梯度；將所述子訓(xùn)練累加梯度同步至所述分布式訓(xùn)練集群的每個(gè)訓(xùn)練節(jié)點(diǎn)。在每個(gè)訓(xùn)練節(jié)點(diǎn)上將計(jì)算不同批量大小訓(xùn)練數(shù)據(jù)的平均梯度，可以保證梯度的震蕩范圍較小，能更準(zhǔn)確的確定梯度下降的方向，加快模型的訓(xùn)練速度，提高模型訓(xùn)練的效率。