模型訓練方法、裝置、存儲介質及設備

基本信息

申請?zhí)?/td> CN202011566357.8 申請日 -
公開（公告）號 CN114757244A 公開（公告）日 2022-07-15
申請公布號 CN114757244A 申請公布日 2022-07-15
分類號 G06K9/62（2022.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 李億;練韻文 申請（專利權）人 華為云計算技術有限公司
代理機構 深圳市賽恩倍吉知識產(chǎn)權代理有限公司 代理人 -
地址 550025貴州省貴安新區(qū)黔中大道交興功路華為云數(shù)據(jù)中心
法律狀態(tài) -

摘要

摘要 本申請公開了一種模型訓練方法、裝置、存儲介質及設備，屬于AI領域。在本申請實施例中，可以獲得神經(jīng)網(wǎng)絡模型對應的梯度信息，然后根據(jù)梯度信息對第一訓練數(shù)據(jù)子集中的訓練數(shù)據(jù)進行評估，獲得評估結果，最后根據(jù)評估結果調整索引表。這樣，調整后的索引表用于在下一個回合獲取第二訓練數(shù)據(jù)子集。也即，本申請實施例可以實現(xiàn)在訓練的過程中根據(jù)梯度信息動態(tài)調整索引表，進而可以在下一回合中根據(jù)索引表讀取相應的第二訓練數(shù)據(jù)子集。通過在每個回合對訓練數(shù)據(jù)進評估，動態(tài)調整訓練過程中的訓練數(shù)據(jù)集，使得在模型的訓練過程中能更快地達到訓練目標，節(jié)約訓練時長和訓練消耗的算力。