买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:深圳市捷易科技有限公司
摘要:本申请提供一种大模型混合并行训练加速方法、装置、设备及存储介质,涉及深度学习技术领域,方法包括:将目标大模型的计算图进行切分,得到多个子计算图;根据各子计算图的计算特征和目标大模型的训练数据分布特征,确定各子计算图的并行策略;结合各子计算图的并行策略、批次大小及流水线阶段数,确定各子计算图在多个计算设备上的执行时间和执行顺序和各子计算图之间的通信时间及通信顺序;根据各子计算图在多个计算设备上的执行时间和执行顺序,以及各子计算图之间的通信时间及通信顺序,生成多个计算设备之间的执行调度方案;根据执行调度方案,在多个计算设备上并行执行子计算图。本申请具有的技术效果是:满足实际应用中大模型高效训练的需求。
主权项:1.一种大模型混合并行训练加速方法,其特征在于,所述方法包括:将目标大模型的计算图进行切分,得到多个子计算图;根据各所述子计算图的计算特征和所述目标大模型的训练数据分布特征,确定各所述子计算图的并行策略;结合各所述子计算图的并行策略、批次大小及流水线阶段数,确定各所述子计算图在多个计算设备上的执行时间和执行顺序,以及各所述子计算图之间的通信时间及通信顺序;根据各所述子计算图在多个计算设备上的执行时间和执行顺序,以及各所述子计算图之间的通信时间及通信顺序,生成所述多个计算设备之间的执行调度方案;根据所述执行调度方案,在所述多个计算设备上并行执行所述子计算图,以使所述目标大模型进行混合并行训练。
全文数据:
权利要求:
百度查询: 深圳市捷易科技有限公司 大模型混合并行训练加速方法、装置、设备及存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。