[發明專利]深度學習訓練任務的管理方法及管理裝置

在審

• 申請號: 202011486962.4
• 申請日: 2020-12-16
• 公開（公告）號: CN112527503A
• 公開（公告）日: 2021-03-19
• 發明（設計）人: 趙明;韓來鵬;陳陽雪;柳笛;杜艷冰
• 申請（專利權）人: 北京地平線信息技術有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50;G06N3/08
• 代理公司: 北京布瑞知識產權代理有限公司 11505
• 代理人: 秦衛中

1.一種深度學習訓練任務的管理方法，包括：

確定第一深度學習訓練任務所在第一任務隊列的資源使用信息；

基于所述第一任務隊列的資源使用信息，調整所述第一任務隊列對應的第一計算機集群的資源容量；

控制調整資源容量后的第一計算機集群執行所述第一深度學習訓練任務。

2.根據權利要求1所述的管理方法，其中，

所述基于所述第一任務隊列的資源使用信息，調整所述第一任務隊列對應的第一計算機集群的資源容量，包括：

確定所述第一任務隊列的資源使用信息中所包括的所述第一任務隊列中的深度學習訓練任務的平均等待時間；

基于所述第一任務隊列中的深度學習訓練任務的平均等待時間與第二任務隊列中的深度學習訓練任務的平均等待時間，將所述第二任務隊列對應的第二計算機集群的部分資源或全部資源的標簽由所述第二計算機集群修改成所述第一計算機集群。

3.根據權利要求2所述的管理方法，其中，在所述將所述第二任務隊列對應的第二計算機集群的部分資源或全部資源的標簽由所述第二計算機集群修改成所述第一計算機集群之后，還包括：基于所述第二任務隊列中的深度學習訓練任務的平均等待時間，將所述部分資源或全部資源的標簽由所述第一計算機集群修改成所述第二計算機集群。

4.根據權利要求1所述的管理方法，還包括：

基于第一預設打分策略確定所述第一深度學習訓練任務的分值；

基于所述第一深度學習訓練任務的分值確定所述第一深度學習訓練任務在所述第一任務隊列中的執行順序；

基于所述執行順序，執行所述控制調整資源容量后的第一計算機集群執行所述第一深度學習訓練任務的步驟。

5.根據權利要求4所述的管理方法，還包括：

基于第二預設打分策略確定第二任務隊列中的第二深度學習訓練任務的分值；

基于所述第二深度學習訓練任務的分值確定所述第二深度學習訓練任務在所述第二任務隊列中的執行順序，其中，所述第一預設打分策略不同于所述第二預設打分策略。

6.根據權利要求4所述的管理方法，其中，所述基于第一預設打分策略確定所述第一深度學習訓練任務的分值，包括：

基于所述第一深度學習訓練任務的提交時間，確定所述第一深度學習訓練任務的第一得分；

基于所述第一深度學習訓練任務的資源申請量，確定所述第一深度學習訓練任務的第二得分；

基于所述第一得分和所述第二得分的加權之和，確定所述第一深度學習訓練任務的分值。

7.根據權利要求4所述的管理方法，其中，所述基于第一預設打分策略確定所述第一深度學習訓練任務的分值，包括：

基于所述第一深度學習訓練任務的提交時間，確定所述第一深度學習訓練任務的第一得分；

基于所述第一深度學習訓練任務的資源申請量，確定所述第一深度學習訓練任務的第二得分；

基于提交所述第一深度學習訓練任務的用戶的身份信息，確定所述第一深度學習訓練任務的第三得分；

基于所述第一得分、所述第二得分以及所述第三得分的加權之和，確定所述第一深度學習訓練任務的分值。

8.一種深度學習訓練任務的管理裝置，包括：

確定模塊，用于確定第一深度學習訓練任務所在第一任務隊列的資源使用信息；

調整模塊，用于基于所述第一任務隊列的資源使用信息，調整所述第一任務隊列對應的第一計算機集群的資源容量；

執行模塊，用于控制調整資源容量后的第一計算機集群執行所述第一深度學習訓練任務。

9.一種計算機可讀存儲介質，所述存儲介質存儲有計算機程序，所述計算機程序用于執行上述權利要求1至7中任一項所述的深度學習訓練任務的管理方法。

10.一種電子設備，所述電子設備包括：

處理器；

用于存儲所述處理器可執行指令的存儲器，

其中，所述處理器用于執行上述權利要求1至7中任一項所述的深度學習訓練任務的管理方法。