买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:苏州元脑智能科技有限公司
摘要:本发明提供一种训练任务检查点文件保存方法、装置、设备及介质,该方法包括:在大语言模型的同步训练保存过程中,对大语言模型当前次迭代的训练任务对应的检查点文件进行解构,得到多个待保存数据,并将待保存数据保存到共享内存中;在异步服务存储过程中,从共享内存中将当前次迭代的训练任务对应的多个待保存数据恢复成检查点文件并存储到共享存储设备中。本发明通过将检查点文件解构成多个待保存数据在共享内存中进行保存,并从共享内存中提取多个待保存数据合并成检查点文件存储在共享存储设备中,避免将检查点文件在训练过程中与训练框架耦合后存储在共享存储设备中,也避免完全了解训练框架机制才能对检查点文件进行保存,具备普适性。
主权项:1.一种训练任务检查点文件保存方法,其特征在于,应用于分布式训练中的训练节点,包括:在大语言模型的同步训练保存过程中,对所述大语言模型当前次迭代的训练任务对应的检查点文件进行解构,得到多个待保存数据,并将待保存数据保存到共享内存中;在大语言模型的异步服务存储过程中,从所述共享内存中将所述当前次迭代的训练任务对应的多个待保存数据恢复成所述检查点文件并存储到共享存储设备中。
全文数据:
权利要求:
百度查询: 苏州元脑智能科技有限公司 训练任务检查点文件保存方法、装置、设备及介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。