[發明專利]一種模型訓練方法及裝置

在審

• 申請號: 202110637088.8
• 申請日: 2021-06-08
• 公開（公告）號: CN113485784A
• 公開（公告）日: 2021-10-08
• 發明（設計）人: 曹云飛;霍龍社;崔煜喆;徐治理;唐雄燕;劉騰飛
• 申請（專利權）人: 中國聯合網絡通信集團有限公司
• 主分類號: G06F9/455
• 分類號: G06F9/455;G06K9/62
• 代理公司: 北京中博世達專利商標代理有限公司 11274
• 代理人: 申健

本發明實施例提供一種模型訓練方法及裝置，涉及機器學習技術領域，能夠保障服務器(包括第一服務器和第二服務器)的運行能力，提升模型訓練的效率。該方法包括：創建目標項目；從數據存儲設備中獲取目標數據集和該目標項目對應的鏡像文件，該目標數據集為所述目標項目對應的數據集；向第二服務器發送該目標數據集和該鏡像文件；接收該第二服務器發送的目標容器對應的互IP地址和端口信息，該目標容器為該鏡像文件對應的容器。

技術領域

本發明實施例涉及機器學習技術領域，尤其涉及一種模型訓練方法及裝置。

背景技術

目前，可以在某一物理機或虛擬機上訓練模型，具體的，以物理機為例，該物理機的本地可以存儲訓練需要的訓練數據，進而該物理機可以基于該訓練數據進行模型訓練，以得到訓練完成的模型。

但是，目前物理機本身的計算資源和存儲資源一般是固定的，如此，上述模型訓練方法可能不能滿足模型訓練所需的相關資源，影響模型的訓練效率。

發明內容

本發明實施例提供一種模型訓練方法及裝置，能夠保障服務器(包括第一服務器和第二服務器)的運行能力，提升模型訓練的效率。

第一方面，本發明實施例提供一種模型訓練方法，包括：創建目標項目；從數據存儲設備中獲取目標數據集和該目標項目對應的鏡像文件，該目標數據集為該目標項目對應的數據集；向第二服務器發送該目標數據集和該鏡像文件；接收該第二服務器發送的目標容器對應的互聯網協議(internet protocol，IP)地址和端口信息，該目標容器為該鏡像文件對應的容器。

第二方面，本發明實施例提供一種模型訓練方法，包括：接收第一服務器發送的目標數據集和目標項目對應的鏡像文件，該目標數據集為該目標項目對應的數據集；基于該目標數據集和該鏡像文件訓練模型，以得到該目標項目對應的訓練模型；向該第一服務器發送目標容器對應的IP地址和端口信息，該目標容器為該鏡像文件對應的容器。

第三方面，本發明實施例提供一種模型訓練方法，包括：接收第一服務器發送的數據集獲取請求消息，該數據集獲取請求消息中包括目標用戶的標識和目標項目的標識，該目標用戶為該目標項目對應的用戶；確定該目標用戶的標識是否為目標數據集對應的用戶標識，或者該目標數據集的數據狀態是否為共享狀態，該目標數據集為該目標項目對應的數據集；在該目標用戶的標識為該目標數據集對應的用戶標識，或者該目標數據集的數據狀態為共享狀態的情況下，向該第一服務器發送該目標數據集。

第四方面，本發明實施例提供一種模型訓練裝置，包括：創建模塊、獲取模塊、發送模塊以及接收模塊；該創建模塊，用于創建目標項目；該獲取模塊，用于從數據存儲設備中獲取目標數據集和該目標項目對應的鏡像文件，該目標數據集為該目標項目對應的數據集；該發送模塊，用于向第二服務器發送該目標數據集和該鏡像文件；該接收模塊，用于接收該第二服務器發送的目標容器對應的IP地址和端口信息，該目標容器為該鏡像文件對應的容器。

第五方面，本發明實施例提供一種模型訓練裝置，包括：接收模塊，訓練模塊以及發送模塊；該接收模塊，用于接收第一服務器發送的目標數據集和目標項目對應的鏡像文件，該目標數據集為該目標項目對應的數據集；該訓練模塊，用于基于該目標數據集和該鏡像文件訓練模型，以得到該目標項目對應的訓練模型；該發送模塊，用于向該第一服務器發送目標容器對應的IP地址和端口信息，該目標容器為該鏡像文件對應的容器。

第六方面，本發明實施例提供一種模型訓練裝置，包括：接收模塊、確定模塊以及發送模塊；該接收模塊，用于接收第一服務器發送的數據集獲取請求消息，該數據集獲取請求消息中包括目標用戶的標識和目標項目的標識，該目標用戶為該目標項目對應的用戶；該確定模塊，用于確定該目標用戶的標識是否為目標數據集對應的用戶標識，或者該目標數據集的數據狀態是否為共享狀態，該目標數據集為該目標項目對應的數據集；該發送模塊，用于在該目標用戶的標識為該目標數據集對應的用戶標識，或者該目標數據集的數據狀態為共享狀態的情況下，向該第一服務器發送該目標數據集。