[發明專利]神經網絡處理方法、計算機系統及存儲介質

有效

• 申請號: 201711483733.5
• 申請日: 2017-12-29
• 公開（公告）號: CN109993287B
• 公開（公告）日: 2019-12-06
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 北京中科寒武紀科技有限公司
• 主分類號: G06N3/063
• 分類號: G06N3/063
• 代理公司: 11606 北京華進京聯知識產權代理有限公司
• 代理人: 王程<國際申請>=<國際公布>=<進入國

1.一種神經網絡的處理方法，其特征在于，所述方法包括如下步驟：

獲取原始網絡的模型數據集和模型結構參數，其中，所述模型數據集包括所述原始網絡中各個計算節點對應的網絡權值，所述模型結構參數包括所述原始網絡中多個計算節點的連接關系；

獲取所述原始網絡中各個計算節點的操作屬性，所述計算節點的操作屬性包括用于表示所述計算節點能夠在專用神經網絡處理器上執行的第一操作屬性和用于表示所述計算節點能夠在通用處理器上執行的第二操作屬性；

根據所述原始網絡的模型數據集、模型結構參數及各個計算節點的操作屬性運行所述原始網絡，獲得所述原始網絡中各個計算節點對應的指令；

若當前計算節點的操作屬性為第一操作屬性時，則將所述當前計算節點對應的網絡權值及指令存儲至第一非易失性存儲器中，以獲得所述原始網絡對應的第一離線模型。

2.根據權利要求1所述的方法，其特征在于，獲取所述原始網絡中各個計算節點的操作屬性的步驟，還包括：

分別判斷各個所述計算節點是否能夠在專用神經網絡處理器上執行；

若當前計算節點能夠在所述專用神經網絡處理器上執行，則將所述當前計算節點標記為第一操作屬性；

若當前計算節點只能在通用處理器上執行，則將所述當前計算節點標記為第二操作屬性。

3.根據權利要求2所述的方法，其特征在于，分別判斷各個所述計算節點是否能夠在專用神經網絡處理器上執行的步驟，還包括：

通過預設函數表查詢所述當前計算節點是否存在等效計算節點，其中，所述等效計算節點為能夠在所述專用神經網絡處理器上執行的計算節點；

若所述當前計算節點存在等效計算節點，則判定所述當前計算節點能夠在所述專用神經網絡處理器上執行；

若所述當前計算節點不存在等效計算節點，則判定所述當前計算節點只能在所述通用處理器上執行。

4.根據權利要求1所述的方法，其特征在于，獲取所述原始網絡中各個計算節點的操作屬性的步驟，包括：

從所述原始網絡的模型數據集或模型結構參數中，獲得所述原始網絡中各個計算節點的操作屬性。

5.根據權利要求2或4所述的方法，其特征在于，所述通用處理器包括中央處理器、圖像處理器、數字信號處理器及現場可編輯門陣列中的一種或多種；

所述第二操作屬性包括CPU操作屬性、GPU操作屬性、DSP操作屬性及FPGA操作屬性的一種或多種。

6.根據權利要求1所述的方法，其特征在于，所述方法還包括如下步驟：

根據所述原始網絡中多個計算節點的連接關系，將兩個以上順序執行的第二計算節點之間的所有第一計算節點等效為一個第一離線節點；其中，所述第一計算節點為具有第一操作屬性的計算節點，所述第二計算節點為具有第二操作屬性的計算節點；所述第一離線模型中還包括所述第一離線節點與所述第二計算節點之間的接口數據。

7.根據權利要求1所述的方法，其特征在于，所述方法還包括如下步驟：

若所述當前計算節點的操作屬性為第二操作屬性時，則將所述當前計算節點對應的網絡權值及指令存儲至第二非易失性存儲器中，以獲得所述原始網絡對應的第二離線模型。

8.根據權利要求7所述的方法，其特征在于，所述方法還包括如下步驟：

根據所述原始網絡中多個計算節點的連接關系，將兩個以上順序執行的第一計算節點之間的所有第二計算節點等效為一個第二離線節點；其中，所述第一計算節點為具有第一操作屬性的計算節點，所述第二計算節點為具有第二操作屬性的計算節點；所述第二離線模型中還包括所述第二離線節點與所述第一計算節點之間的接口數據。