公开(公告)号：US20250124347A1

公开(公告)日：2025-04-17

申请号：US18804240

申请日：2024-08-14

Applicant: ZHEJIANG LAB

Inventor： Fei YANG ,  Shuang PENG ,  Ning SUN ,  Fangyu WANG ,  Aimin PAN

IPC: G06N20/00

Abstract: A training model allocation method, an apparatus, a computer device, and a storage medium are provided. The method includes: acquiring hierarchy information, calculating parameter information, and a training data set of a to-be-trained model; dividing the to-be-trained model into sub-models according to the hierarchy information, and allocating each of sub-models to machine nodes in a training cluster; dividing each of sub-models into sub-model slices according to the calculating parameter information, and allocating each of sub-model slices to computing processors of the machine nodes in the training cluster; dividing the training data set into training data subsets according to the calculating parameter information, and allocating each of training data subsets to the computing processors in the training cluster; and training the to-be-trained model according to all computing processors, sub-model slices, and training data subsets in the training cluster.