專為大語(yǔ)言模型訓(xùn)練定制 微軟推出高效FP8混合精度訓(xùn)練框架

11月10日消息,來(lái)自微軟 Azure 和微軟研究院的一組研究人員推出了一個(gè)高效的 FP8 混合精度框架,專為大型語(yǔ)言模型訓(xùn)練量身定制。

經(jīng)過(guò)測(cè)試,與廣泛采用的 BF16 混合精度方法相比,F(xiàn)P8 混合精度框架內(nèi)存占用減少 27% 至 42%,權(quán)重梯度通信開(kāi)銷(xiāo)顯著降低 63% 至 65%。

運(yùn)行速度比廣泛采用的 BF16 框架(例如 Megatron-LM)快了 64%,比英偉達(dá) Transformer Engine 的速度快了 17%。

在訓(xùn)練 GPT-175B 模型時(shí),混合 FP8 精度框架在 H100 GPU 平臺(tái)上節(jié)省 21% 的內(nèi)存,而且相比較 TE(Transformer Engine),訓(xùn)練時(shí)間減少 17%。

免責(zé)聲明:本網(wǎng)站內(nèi)容主要來(lái)自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準(zhǔn)確性及可靠性,但不保證有關(guān)資料的準(zhǔn)確性及可靠性,讀者在使用前請(qǐng)進(jìn)一步核實(shí),并對(duì)任何自主決定的行為負(fù)責(zé)。本網(wǎng)站對(duì)有關(guān)資料所引致的錯(cuò)誤、不確或遺漏,概不負(fù)任何法律責(zé)任。任何單位或個(gè)人認(rèn)為本網(wǎng)站中的網(wǎng)頁(yè)或鏈接內(nèi)容可能涉嫌侵犯其知識(shí)產(chǎn)權(quán)或存在不實(shí)內(nèi)容時(shí),應(yīng)及時(shí)向本網(wǎng)站提出書(shū)面權(quán)利通知或不實(shí)情況說(shuō)明,并提供身份證明、權(quán)屬證明及詳細(xì)侵權(quán)或不實(shí)情況證明。本網(wǎng)站在收到上述法律文件后,將會(huì)依法盡快聯(lián)系相關(guān)文章源頭核實(shí),溝通刪除相關(guān)內(nèi)容或斷開(kāi)相關(guān)鏈接。

2023-11-10
專為大語(yǔ)言模型訓(xùn)練定制 微軟推出高效FP8混合精度訓(xùn)練框架
內(nèi)存占用少42%。

長(zhǎng)按掃碼 閱讀全文