[發(fā)明專利]通用機器學(xué)習(xí)模型文件解析方法、裝置、存儲介質(zhì)

有效

• 申請?zhí)枺?/td>: 201811459853.6
• 申請日: 2018-11-30
• 公開（公告）號: CN111260071B
• 公開（公告）日: 2022-04-08
• 發(fā)明（設(shè)計）人: 不公告發(fā)明人
• 申請（專利權(quán)）人: 上海寒武紀信息科技有限公司
• 主分類號: G06N20/00
• 分類號: G06N20/00
• 代理公司: 北京華進京聯(lián)知識產(chǎn)權(quán)代理有限公司 11606
• 代理人: 孫巖

1.一種通用機器學(xué)習(xí)模型文件解析方法，其特征在于，所述通用機器學(xué)習(xí)模型文件包括目標通用機器學(xué)習(xí)模型、模型目錄，所述方法包括：

獲取通用機器學(xué)習(xí)模型文件的文件標識碼；

檢測所述文件標識碼是否符合預(yù)設(shè)規(guī)則，若所述文件標識碼符合預(yù)設(shè)規(guī)則，則在所述通用機器學(xué)習(xí)模型文件中讀取所述模型目錄；所述模型目錄為通用機器學(xué)習(xí)模型文件中所有模型存放位置的記錄；所述預(yù)設(shè)規(guī)則表示通用機器學(xué)習(xí)模型文件的標識碼的描述信息；其中，通用機器學(xué)習(xí)模型文件的標識碼是指通用機器學(xué)習(xí)模型文件附帶的起標識作用的字符，通過識別文件的標識碼可以將不同的通用機器學(xué)習(xí)模型文件區(qū)分開，便于準確獲取對應(yīng)的通用機器學(xué)習(xí)模型文件；

根據(jù)所述模型目錄，在所述通用機器學(xué)習(xí)模型文件中讀取所述目標通用機器學(xué)習(xí)模型。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述若所述文件標識碼符合預(yù)設(shè)規(guī)則，則在所述通用機器學(xué)習(xí)模型文件中讀取模型目錄包括：

獲取所述通用機器學(xué)習(xí)模型文件的校驗碼；

校驗所述校驗碼與預(yù)設(shè)標準碼是否一致，若所述校驗碼與預(yù)設(shè)標準碼不一致，則執(zhí)行糾錯運算。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述校驗所述校驗碼與預(yù)設(shè)標準碼是否一致，若所述校驗碼與預(yù)設(shè)標準碼不一致，則執(zhí)行糾錯運算包括：

獲取糾錯碼；

根據(jù)所述糾錯碼對所述通用機器學(xué)習(xí)模型文件進行糾錯，得到糾錯后的模型文件；

校驗所述糾錯后的模型文件的校驗碼與所述預(yù)設(shè)標準碼是否一致；

若所述糾錯后的通用機器學(xué)習(xí)模型文件的校驗碼與所述預(yù)設(shè)標準碼一致，則在所述通用機器學(xué)習(xí)模型文件中讀取模型目錄。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述根據(jù)所述模型目錄，讀取對應(yīng)的通用機器學(xué)習(xí)模型包括：

獲取目標通用機器學(xué)習(xí)模型的在所述通用機器學(xué)習(xí)模型文件中的存儲偏移量；

根據(jù)所述存儲偏移量，讀取所述目標通用機器學(xué)習(xí)模型。

5.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述方法還包括：

讀取所述通用機器學(xué)習(xí)模型中的硬件參數(shù)信息；

根據(jù)所述硬件參數(shù)信息，生成硬件匹配信息。

6.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述方法還包括：

對所述通用機器學(xué)習(xí)模型中可共享的數(shù)據(jù)進行拆解，得到棧區(qū)數(shù)據(jù)，將所述通用機器學(xué)習(xí)模型中不可共享的數(shù)據(jù)進行拆解，得到堆區(qū)數(shù)據(jù)；

根據(jù)所述棧區(qū)數(shù)據(jù)、所述堆區(qū)數(shù)據(jù)以及輸入數(shù)據(jù)計算，得到輸出數(shù)據(jù)。

7.一種通用機器學(xué)習(xí)模型文件解析裝置，其特征在于，所述裝置包括：

文件獲取器，目錄解析器以及模型讀取器；所述目錄解析器分別與所述文件獲取器、以及所述模型讀取器相連；

所述文件獲取器，用于獲取通用機器學(xué)習(xí)模型文件；所述通用機器學(xué)習(xí)模型文件包括目標通用機器學(xué)習(xí)模型、模型目錄；

所述文件獲取器包括文件頭校驗器，用于獲取所述通用機器學(xué)習(xí)模型文件的文件標識碼；

所述目錄解析器，用于檢測所述文件標識碼是否符合預(yù)設(shè)規(guī)則，若所述文件標識碼符合預(yù)設(shè)規(guī)則，則在所述通用機器學(xué)習(xí)模型文件中讀取模型目錄；所述模型目錄為通用機器學(xué)習(xí)模型文件中所有模型存放位置的記錄；所述預(yù)設(shè)規(guī)則表示通用機器學(xué)習(xí)模型文件的標識碼的描述信息；其中，通用機器學(xué)習(xí)模型文件的標識碼是指通用機器學(xué)習(xí)模型文件附帶的起標識作用的字符，通過識別文件的標識碼可以將不同的通用機器學(xué)習(xí)模型文件區(qū)分開，便于準確獲取對應(yīng)的通用機器學(xué)習(xí)模型文件；

所述模型讀取器，用于根據(jù)所述模型目錄，在通用機器學(xué)習(xí)模型文件中讀取所述目標通用機器學(xué)習(xí)模型。

8.根據(jù)權(quán)利要求7所述的裝置，其特征在于，

所述文件頭校驗器還用于：

獲取所述通用機器學(xué)習(xí)模型文件的校驗碼；以及校驗所述校驗碼與預(yù)設(shè)標準碼是否一致，若所述校驗碼與預(yù)設(shè)標準碼不一致，則執(zhí)行糾錯運算。