Invention Grant
- Patent Title: 训练模型分配方法、装置、计算机设备和存储介质
-
Application No.: CN202311336127.6Application Date: 2023-10-16
-
Publication No.: CN117112145BPublication Date: 2024-02-13
- Inventor: 杨非 , 彭爽 , 孙柠 , 汪芳羽 , 谈科 , 潘爱民
- Applicant: 之江实验室
- Applicant Address: 浙江省杭州市余杭区中泰街道科创大道之江实验室
- Assignee: 之江实验室
- Current Assignee: 之江实验室
- Current Assignee Address: 浙江省杭州市余杭区中泰街道科创大道之江实验室
- Agency: 杭州华进联浙知识产权代理有限公司
- Agent 吴婷婷
- Main IPC: G06F9/455
- IPC: G06F9/455 ; G06F18/214 ; G06N3/0464 ; G06N3/08
Abstract:
本申请涉及一种训练模型分配方法、装置、计算机设备和存储介质。所述方法包括:获取待训练模型的模型信息和训练数据集;将训练模型根据层级信息划分为至少两个子模型,并将各子模型分配至训练集群中各机器节点;将各子模型根据计算参数信息划分为至少两个子模型切片,并将各子模型切片分配至训练集群中各机器节点的各计算处理器;将训练数据集根据计算参数信息划分为至少两个训练子数据集,并将各训练子数据集分配至训练集群中各计算处理器;根据训练集群中所有计算处理器,以及所有计算处理器对应的子模型切片和训练数据子集,对待训练模型进行训练。采用本申请的方法能够提高模型训练效率。
Public/Granted literature
- CN117112145A 训练模型分配方法、装置、计算机设备和存储介质 Public/Granted day:2023-11-24
Information query