模型训练方法及装置、电子设备、存储介质和程序产品

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
模型训练方法及装置、电子设备、存储介质和程序产品
申请号:CN202511447519
申请日期:2025-10-11
公开号:CN120930831B
公开日期:2025-12-05
类型:发明专利
摘要
本公开涉及一种模型训练方法及装置、电子设备、存储介质和程序产品,涉及计算机技术领域,所述方法包括:将待训练模型切分为N个部分分别分配至N个第二处理器,每个第二处理器对应的流水线用于训练待训练模型的不同部分;协同调度第一处理器和N个第二处理器对待训练模型进行流水线并行训练,得到训练好的目标模型;其中,第一处理器用于与第1个第二处理器共同训练待训练模型的第1部分,以及与第N个第二处理器共同训练待训练模型的第N部分,以使每个第二处理器对待训练模型不同部分的训练时间相同。本公开实施例可在保持待训练模型的完整性和兼容性条件下,消除或减少各第二处理器的空闲等待时间,提高模型训练的效率。
技术关键词
处理器 计算机程序指令 异构系统 模型训练方法 流水线 分布式训练 模型训练装置 调度器 电子设备 计算机程序产品 可读存储介质 存储器 数据 动态 模块